溫馨提示×

python assert如何驗證輸入

小樊
81
2024-11-15 02:17:41
欄目: 編程語言

在Python中,assert語句用于檢查一個條件是否為真。如果條件為假,程序?qū)⒁l(fā)AssertionError異常。要使用assert驗證輸入,你可以將輸入作為參數(shù)傳遞給assert語句,然后檢查該輸入是否滿足預(yù)期的條件。

以下是一個簡單的示例,展示了如何使用assert驗證輸入是否為整數(shù):

def process_input(input_value):
    assert isinstance(input_value, int), f"Input must be an integer, but got {type(input_value).__name__}"
    
    # 如果輸入是整數(shù),繼續(xù)處理
    print(f"Processing input: {input_value}")

# 測試函數(shù)
process_input(42)  # 輸出: Processing input: 42
process_input("hello")  # 引發(fā) AssertionError: Input must be an integer, but got str

在這個例子中,我們使用assert語句檢查input_value是否為整數(shù)。如果不是整數(shù),程序?qū)⒁l(fā)一個帶有錯誤消息的AssertionError異常。如果輸入是整數(shù),程序?qū)⒗^續(xù)處理輸入。

0