Python 里assert语句的作用和用法|极客教程

Python 里assert语句的作用和用法

在Python中，assert语句用于检查条件是否为真，如果条件为假，则会引发AssertionError异常。assert语句通常用于在程序中检查代码的正确性、可用性和调试。

assert语句的语法

assert语句的语法如下：

assert condition, "error message"

其中condition是要检查的条件，如果条件为假，则会引发AssertionError异常并输出指定的错误消息"error message"。

assert语句的使用场景

检查函数参数：

def divide(x, y):
    assert y != 0, "Divider cannot be zero"
    return x / y

print(divide(10, 2))  # 输出：5.0
print(divide(10, 0))  # 输出：AssertionError: Divider cannot be zero

检查数据的有效性：

def calculate_area(length, width):
    assert length > 0 and width > 0, "Length and width must be positive numbers"
    return length * width

print(calculate_area(5, 10))  # 输出：50
print(calculate_area(-5, 10))  # 输出：AssertionError: Length and width must be positive numbers

检查函数返回值：

def divide(x, y):
    result = x / y
    assert y != 0, "Divider cannot be zero"
    return result

print(divide(10, 2))  # 输出：5.0
print(divide(10, 0))  # 输出：AssertionError: Divider cannot be zero

assert语句的注意事项

assert语句通常用于调试和测试阶段，对于生产环境的代码，应该避免使用assert语句，因为默认情况下，Python解释器会忽略assert语句。
可以通过命令行参数-O或者在代码中设置PYTHONOPTIMIZE环境变量来禁用assert语句。

示例代码运行结果

运行以下示例代码：

def divide(x, y):
    assert y != 0, "Divider cannot be zero"
    return x / y

print(divide(10, 2))  # 正确结果：5.0
print(divide(10, 0))  # 错误结果：AssertionError: Divider cannot be zero

第一个print语句输出了正确的结果5.0，而第二个print语句则触发了AssertionError异常并输出了错误消息"Divider cannot be zero"。