Python 断言(assert)|极客教程

Python 断言(assert)

在编程中，断言（assert）是一种用于检查程序逻辑是否正确的方法。通过使用断言，程序员可以在代码中插入检查点，确保某个条件为真，否则会引发 Assertion 错误。当断言条件为假时，程序会立即停止执行，并抛出 AssertionError 异常。

为什么使用断言？

断言可以用来验证程序在运行时是否满足特定条件。通过在代码中插入断言语句，程序员可以在开发阶段或者调试阶段，快速捕获程序中的错误。这有利于提高代码质量和减少调试时间。

在软件开发中，断言通常用于以下情况：

检查函数的输入参数是否符合要求
确保程序执行过程中的状态满足预期
防止空指针引用和数组越界等错误

如何使用断言？

在 Python 中，使用关键字 assert 来创建断言语句。assert 后面跟着一个条件表达式，当该条件为假时，会触发断言错误。

示例1：检查函数参数是否为正数

def divide(a, b):
    assert b != 0, "除数不能为0"
    assert a > 0 and b > 0, "除数和被除数都必须为正数"
    return a / b

print(divide(10, 2))  # 正确情况
print(divide(10, 0))  # 除数为0，触发 AssertionError
print(divide(-10, 2))  # 被除数为负数，触发 AssertionError

运行结果：

5.0
AssertionError: 除数不能为0
AssertionError: 除数和被除数都必须为正数

示例2：验证程序执行状态

balance = 1000
withdraw_amount = 1500

assert withdraw_amount <= balance, "提现金额超过账户余额"

balance -= withdraw_amount
print("提现成功，余额为：", balance)

运行结果：

AssertionError: 提现金额超过账户余额

断言的注意事项

使用断言时，需要注意以下几点：
1. 断言主要用于排除程序中的错误情况，而不是处理异常情况。如果某个条件可能是异常但是不是错误，应该使用 try-except 来处理。
2. 断言可以通过 -O 参数关闭，即 python -O your_script.py，关闭后所有的断言将不会触发，这样可以提高程序的执行效率。
3. 尽量避免在断言中添加复杂的逻辑操作，断言应该简洁明了，只用于验证基本的条件是否为真。