Python断言使用教程_assert在开发中的应用

assert是Python中仅用于开发调试的断言工具，检查条件为假时抛出AssertionError，加-O参数会忽略；适用于验证内部逻辑不变式，不可用于用户输入校验或生产环境异常处理。

断言（assert）是 Python 中用于调试和开发阶段快速验证假设的简洁工具，它不是错误处理机制，也不该用于用户输入校验或生产环境的异常控制。

assert 的基本语法和运行逻辑

assert 语句检查一个表达式是否为真；如果为假，立即抛出 AssertionError 异常，并可附带可选的提示信息：

assert condition, "提示消息"

例如：

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

当 b == 0 时，程序中断并报错，帮助开发者快速定位逻辑错误。

注意：Python 在运行时加上 -O（优化）参数会全局忽略所有 assert 语句，所以它只在开发/测试阶段有效，不会影响线上性能。

适合用 assert 的典型场景

断言适用于“本不该发生，一旦发生就说明代码有严重逻辑缺陷”的情况：

• 函数输入参数满足内部约定（如“该函数只接收正整数”，但不代替用户输入校验）
• 算法中间状态检查（如循环不变式、递归终止条件前的状态）
• 私有方法的前置/后置条件（例如“调用此方法前，对象必须已初始化”）
• 调试时临时加的“这里值一定不为空”、“这个列表长度应始终 ≥ 2”等假设验证

assert 不该用在哪里

滥用 assert 可能掩盖问题或导致线上行为不可控：

• 不用于校验用户输入或外部数据：比如表单提交、API 请求参数——应使用 if + 显式异常（如 ValueError）并给出友好提示
• 不替代 try/except：它不提供恢复逻辑，无法捕获和处理，仅用于中止
• 不用于资源检查（如文件是否存在、网络连通）：这类检查可能随环境变化，而 assert 应只依赖代码逻辑本身
• 不写可能改变状态的表达式：如 assert do_something() == True，因为加 -O 后该调用会被跳过，引发副作用不一致

与单元测试中的断言协同使用

在 unittest 或 pytest 中，测试用例里大量使用 self.assertEqual()、assert x in y 等——这些是测试框架提供的断言方法，和内置 assert 语句本质相同，但更丰富、可读性更强、支持详细失败报告。

你可以把函数内部的 assert 看作“自检开关”，把测试文件里的断言看作“外部验收标准”。两者互补：内部 assert 帮你守住模块边界，测试断言帮你守住接口契约。

例如，在实现一个栈类时，可在 pop() 方法内加 assert self._items 防止空栈弹出；同时在测试中写 with self.assertRaises(IndexError): stack.pop() 来验证对外行为是否符合预期。