assert 是 Python 中的一種調(diào)試工具，用于在代碼中設(shè)置斷言（assertion）。斷言是一種聲明，用于確保某個條件為真。如果條件為假，assert 會觸發(fā)一個 AssertionError 異常，并可選地輸出錯誤信息。

語法

assert condition, message

• condition: 需要檢查的條件表達(dá)式。如果條件為 False，則觸發(fā)異常。
• message（可選）: 當(dāng)條件為 False 時，輸出的錯誤信息。如果未提供，則默認(rèn)輸出 AssertionError。

工作原理

1. 如果 condition 為 True，程序繼續(xù)正常執(zhí)行。
2. 如果 condition 為 False，則拋出 AssertionError，并輸出 message（如果提供了）。

示例

1. 基本用法

x = 10
assert x == 10  # 條件為 True，程序繼續(xù)執(zhí)行
assert x > 20   # 條件為 False，觸發(fā) AssertionError

2. 帶錯誤信息

x = 5
assert x > 10, "x 必須大于 10"

運行上述代碼會拋出：

AssertionError: x 必須大于 10

3. 用于函數(shù)參數(shù)檢查

def divide(a, b):assert b != 0, "除數(shù)不能為 0"return a / bprint(divide(10, 2))  # 輸出: 5.0
print(divide(10, 0))  # 觸發(fā) AssertionError: 除數(shù)不能為 0

4. 用于調(diào)試復(fù)雜邏輯

def calculate_discount(price, discount):assert 0 <= discount <= 1, "折扣必須在 0 到 1 之間"return price * (1 - discount)print(calculate_discount(100, 0.2))  # 輸出: 80.0
print(calculate_discount(100, 1.5))  # 觸發(fā) AssertionError: 折扣必須在 0 到 1 之間

注意事項

1. 調(diào)試用途：assert 主要用于調(diào)試和開發(fā)階段，用于捕捉不應(yīng)該發(fā)生的邏輯錯誤。它不應(yīng)該用于處理用戶輸入或外部數(shù)據(jù)驗證，因為這些情況應(yīng)該使用 if 語句和異常處理。

2. 性能影響：在生產(chǎn)環(huán)境中，可以通過運行 Python 時加上 -O（優(yōu)化）選項來禁用所有 assert 語句。例如：

   python -O script.py
   

   這樣所有 assert 語句都會被忽略。

3. 替代方案：如果需要更健壯的輸入驗證或錯誤處理，建議使用 if 語句和 raise 顯式拋出異常。例如：

   if b == 0:raise ValueError("除數(shù)不能為 0")
   

總結(jié)

assert 是一種簡單而有效的調(diào)試工具，用于確保代碼中的某些條件為真。它適合在開發(fā)和測試階段使用，但不適合用于生產(chǎn)環(huán)境中的輸入驗證或錯誤處理。