是的,Python 的 assert
語句可以用于驗證函數的返回值。assert
語句會在其條件為假時引發 AssertionError
異常。你可以使用 assert
語句來檢查函數的返回值是否符合預期。
例如,假設你有一個函數 add
,它接受兩個參數并返回它們的和:
def add(a, b):
return a + b
你可以使用 assert
語句來驗證 add
函數的返回值是否正確:
assert add(1, 2) == 3, "add(1, 2) should return 3"
在這個例子中,如果 add(1, 2)
的返回值不等于 3,assert
語句將引發一個 AssertionError
異常,并顯示提供的錯誤消息。
請注意,assert
語句通常用于開發和測試階段,以確保代碼的正確性。在生產環境中,你可能希望使用更健壯的錯誤處理機制,例如拋出異常或使用日志記錄。