在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
异常。如果输入是整数,程序将继续处理输入。