在Python中,assert
語句用于檢查一個條件是否為真。如果條件為假,程序將引發AssertionError
異常。要使用assert
驗證輸入,你可以將輸入作為參數傳遞給assert
語句,然后檢查該輸入是否滿足預期的條件。
以下是一個簡單的示例,展示了如何使用assert
驗證輸入是否為整數:
def process_input(input_value):
assert isinstance(input_value, int), f"Input must be an integer, but got {type(input_value).__name__}"
# 如果輸入是整數,繼續處理
print(f"Processing input: {input_value}")
# 測試函數
process_input(42) # 輸出: Processing input: 42
process_input("hello") # 引發 AssertionError: Input must be an integer, but got str
在這個例子中,我們使用assert
語句檢查input_value
是否為整數。如果不是整數,程序將引發一個帶有錯誤消息的AssertionError
異常。如果輸入是整數,程序將繼續處理輸入。