是的,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 語句通常用于開發和測試階段,以確保代碼的正確性。在生產環境中,你可能希望使用更健壯的錯誤處理機制,例如拋出異?;蚴褂萌罩居涗?。