断言基础：什么是断言、断言的作用、assert语句的基本语法
大家好，我是你们的老朋友。今天咱们聊聊断言——这个在调试代码时经常被忽视，但用好了能让你少掉不少头发的工具。
说实话，我刚开始写代码那会儿，对断言的态度就是「可有可无」。直到有一次，一个线上bug让我排查了整整两天，最后发现是一个不该为空的参数传了None进来。从那以后，我养成了一个习惯：任何可能出问题的地方，先写个断言兜底。
什么是断言？
断言，说白了就是「我坚信这里一定是这样」。你在代码里写一句断言，相当于对程序说：
「哥们儿，如果这个条件不成立，你就别往下跑了，赶紧报错让我看看。」
它不是一个复杂的机制。就是一个简单的检查点。你想想看，我们写代码时经常会有一些「隐含的假设」——比如这个列表一定不为空、那个参数一定是正数、这个字典一定有某个键。断言就是把这些假设显式地写出来。
核心定义：断言是一种在程序运行时检查布尔表达式的机制。如果表达式为真，程序继续执行；如果为假，程序抛出 AssertionError 异常并终止。
断言的作用
我总结了一下，断言主要有三个作用。嗯，这三个作用我在实际项目里都踩过坑，所以印象特别深。
- 调试辅助：快速定位逻辑错误。我在重构一个老项目时，发现一个函数返回值总是异常。加了几行断言后，立刻发现是某个中间状态被意外修改了。如果没有断言，我可能要在几十个函数调用里慢慢找。
- 契约检查：确保函数调用者遵守约定。比如一个函数要求传入的列表已经排好序，你可以在函数开头写个断言。我曾经在一个数据处理管道里这么干过，效果拔群——谁传了乱序数据，立刻就能抓到。
- 文档化假设：断言本身就是一种可执行的文档。它比注释更可靠，因为注释可能过时，但断言会一直运行检查。我习惯在关键逻辑前写断言，这样后来者一看就知道「哦，这里作者假设了什么」。
我的个人经验：断言最适合检查「不应该发生」的情况。比如一个函数内部的状态机状态值，理论上只可能是0、1、2。如果出现了3，那一定是代码有bug。这种场景用断言，比用if判断然后打日志要干净得多。
assert语句的基本语法
Python里的断言语法非常简单，就一行：
assert 条件表达式 [, 错误信息]来，看几个例子：
# 最基本的用法
assert x > 0
# 带错误信息的用法
assert x > 0, "x必须大于0，当前值为{}".format(x)
# 检查列表不为空
assert len(data) > 0, "数据列表不能为空"
# 检查类型
assert isinstance(user_id, int), "用户ID必须是整数类型"你可能会问：这和if判断有什么区别？
区别大了。if判断是正常的业务逻辑，处理的是「可能发生」的情况。而断言处理的是「理论上不应该发生」的情况。举个例子：
# 这是正常的业务逻辑——用户可能输入负数
def calculate_square_root(x):
if x < 0:
return None # 或者抛出一个业务异常
return math.sqrt(x)
# 这是断言——内部状态不应该出现负数
def process_internal_state(state):
assert state.value >= 0, "内部状态值异常：{}".format(state.value)
# 继续处理...注意：断言在Python优化模式（python -O）下会被完全忽略。所以千万不要用断言来做数据验证或安全检查！我曾经见过有人用断言检查用户权限，结果上线时加了-O参数，所有权限检查都失效了。那场面，啧啧。
断言的使用场景总结
我整理了一个表格，方便你快速对照：
| 场景 | 推荐使用断言？ | 原因 |
|---|---|---|
| 检查函数参数合法性（内部函数） | ✅ 是 | 内部调用者应该遵守约定 |
| 检查用户输入 | ❌ 否 | 用户输入不可信，且可能被优化掉 |
| 检查中间计算结果 | ✅ 是 | 快速定位逻辑错误 |
| 检查文件/网络资源是否存在 | ❌ 否 | 外部环境不可控，应该用异常处理 |
| 单元测试中的验证 | ✅ 是 | 测试环境可控，断言很合适 |
| 安全检查/权限验证 | ❌ 否 | 不能依赖断言做安全控制 |
本章知识体系
下面这张图，是我自己画的。它把断言的核心知识点串在了一起，方便你理解整体结构：
这张图把本章的三个核心知识点串在了一起。你可以看到，断言的核心就是「运行时检查布尔表达式，失败时抛出AssertionError」。它的三个作用——调试辅助、契约检查、文档化假设——在实际开发中各有各的用武之地。而语法虽然简单，但那个「-O模式失效」的坑，你一定要记住。
一个小技巧：我习惯在写复杂函数时，先写断言再写逻辑。比如「这个参数不能为None」、「这个列表长度必须大于0」。先把假设写清楚，再写实现。这样代码写完后，调试起来特别顺畅——因为一旦假设被打破，程序会立刻告诉你，而不是在某个角落里静悄悄地出错。
好了，断言的基础就聊到这儿。记住一句话：断言不是用来处理错误的，而是用来发现错误的。用好它，你的代码会变得更健壮，调试也会变得更轻松。
可可精品资料库