Python assert 練習：用斷言測試函式

寫出預期結果

assert add(2, 3) == 5 把預期變成可執行程式碼。實現變化時，檢查會在確切的行失敗。每個斷言只關注一個行為，失敗原因就更容易診斷。

def add(a, b):
    return a + b
assert add(2, 3) == 5
print("ok")

一個清楚的聲明

加上說明差距的訊息

斷言失敗時會顯示逗號後的訊息。寫清預期值，必要時也寫實際輸入。避免只寫 failed 之類的訊息，它只是重複例外已經表達的內容。

def double(n):
    return n * 2
try:
    assert double(3) == 7, "double(3) should be 6"
except AssertionError as error:
    print(error)

有用的失敗訊息

捕獲失敗來研究它

AssertionError 是普通例外，小範例可以捕獲它並列印訊息。真實測試中應讓失敗傳給測試執行器。把失敗藏起來，就失去了檢查的意義。

try:
    assert 1 == 2, "one is not two"
except AssertionError as error:
    print(error)

查看失敗，而不是掩蓋它

同時測試正常值和邊界值

對 2 和 3 正常的函式，遇到 0、空列表或負數仍可能出錯。每組小測試至少加一個邊界情況。斷言讓這些情況成為程式碼的一部分，而不是一句註解。

def total(values):
    return sum(values)
assert total([]) == 0
assert total([1, 2, 3]) == 6
print("boundary and normal cases pass")

加入一個邊界情況

知道 assert 不適合什麼

不要用 assert 驗證密碼、權限或使用者輸入。Python 的 -O 參數會移除 assert 語句，用這種方式寫的安全檢查可能直接消失。驗證時應拋出 ValueError 或自定義例外。

def positive(n):
    if n <= 0:
        raise ValueError("n must be positive")
    return n
print(positive(3))

驗證應該寫在 if 裡

你的練習清單

• 測試和內部不變數使用 assert。
• 加上說明預期行為的訊息。
• 至少測試一個邊界情況。
• 只有研究失敗時才捕獲 AssertionError。
• 使用者輸入驗證用明確例外，不要用 assert。

常見問題

繼續閱讀：

• Python assert 測試
• Python 練習題帶答案：如何練習而不只是抄答案
• Python 初學者練習計畫：4 週上手路線