温馨提示×

python assert如何验证输入

小樊
81
2024-11-15 02:17:41
栏目: 编程语言

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

0