在 Python 中,assert
語句用于在代碼中插入調試斷言
以下是一個示例,展示了如何自定義 assert
函數的行為:
import sys
def custom_assert(condition, message):
if not condition:
# 在這里自定義行為,例如記錄錯誤、發送通知等。
print(f"自定義斷言失敗: {message}")
# 結束程序執行
sys.exit(1)
# 使用自定義斷言
custom_assert(1 == 2, "1 不等于 2")
在這個示例中,我們創建了一個名為 custom_assert
的函數,該函數接受兩個參數:condition
和 message
。當條件不滿足時,它會打印一條自定義錯誤消息并終止程序執行。你可以根據需要修改此函數以實現所需的自定義行為。
請注意,這種方法不會覆蓋或更改原始 assert
語句的功能。要使用自定義斷言,需要在代碼中顯式調用 custom_assert()
函數。