python中的assert方法

assert 是 Python 中的一个调试工具，用于检查某个条件是否为真。如果条件为真，程序继续执行；如果条件为假，assert 会抛出 AssertionError 异常，并可选地输出一条错误信息。

语法

assert condition, message

• condition：要检查的条件表达式。
• message（可选）：当条件为假时，输出的错误信息。

经典例子

1. 检查变量值

def divide(a, b):
    assert b != 0, "除数不能为零"
    return a / b

print(divide(10, 2))  # 输出: 5.0
print(divide(10, 0))  # 抛出 AssertionError: 除数不能为零

在这个例子中，assert 确保除数 b 不为零。如果 b 为零，程序会抛出 AssertionError 并输出错误信息。

2. 检查列表是否为空

def get_first_element(lst):
    assert len(lst) > 0, "列表不能为空"
    return lst[0]

print(get_first_element([1, 2, 3]))  # 输出: 1
print(get_first_element([]))         # 抛出 AssertionError: 列表不能为空

这里 assert 确保列表 lst 不为空。如果为空，程序会抛出异常。

3. 检查函数参数类型

def add_numbers(a, b):
    assert isinstance(a, int), "a 必须是整数"
    assert isinstance(b, int), "b 必须是整数"
    return a + b

print(add_numbers(2, 3))  # 输出: 5
print(add_numbers(2, "3"))  # 抛出 AssertionError: b 必须是整数

在这个例子中，assert 确保 a 和 b 都是整数。如果不是，程序会抛出异常。

4. 检查函数返回值

def is_positive(n):
    return n > 0

def process_positive_number(n):
    assert is_positive(n), "输入必须是正数"
    return n * 2

print(process_positive_number(5))  # 输出: 10
print(process_positive_number(-1))  # 抛出 AssertionError: 输入必须是正数

这里 assert 确保输入 n 是正数。如果不是，程序会抛出异常。

注意事项

• assert 主要用于调试和开发阶段，帮助开发者快速发现代码中的问题。
• 在生产环境中，assert 语句可能会被禁用（通过 -O 或 -OO 选项），因此不应依赖 assert 来处理关键的错误检查。

总结

assert 是一个简单而强大的工具，用于在代码中插入检查点，确保某些条件在运行时为真。它可以帮助你快速发现和修复代码中的错误。