在Python中,assert
語句用于檢查一個條件是否為真。如果條件為假,程序將引發AssertionError
異常。要使用assert
處理數據類型,你可以在條件中檢查變量的數據類型。這是一個示例:
def process_data(data):
# 檢查輸入數據是否為整數
assert isinstance(data, int), f"Expected an integer, but got {type(data).__name__}"
# 對數據進行一些處理
result = data * 2
return result
# 正確的用法
try:
print(process_data(5)) # 輸出:10
except AssertionError as e:
print(e)
# 錯誤的用法,將引發AssertionError異常
try:
print(process_data(3.14)) # 輸出:Expected an integer, but got float
except AssertionError as e:
print(e)
在這個示例中,我們定義了一個名為process_data
的函數,該函數接受一個參數data
。在函數內部,我們使用assert
語句檢查data
是否為整數。如果不是整數,程序將引發AssertionError
異常,并顯示一條錯誤消息,指出期望的數據類型和實際的數據類型。