Python内置断言：assert语句的用法、优缺点、适用场景

说到断言，很多新手觉得它就是个「调试用的print替代品」。嗯，这么说也对，但不够准确。我个人的理解是：assert是你写给未来自己的契约——它告诉读代码的人「这里的情况必须是这样，否则就是bug」。

Python的assert语句，说白了就是一行代码搞定条件检查+异常抛出。咱们先看看它长什么样。

1. assert的基本用法

语法极其简单：

assert 条件表达式
# 等价于：
if not 条件表达式:
    raise AssertionError

也可以带错误信息：

assert 条件表达式, "错误描述信息"
# 等价于：
if not 条件表达式:
    raise AssertionError("错误描述信息")

来个实际例子：

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

# 正常调用
print(divide(10, 2))  # 输出 5.0

# 触发断言
print(divide(10, 0))  # AssertionError: 除数不能为0

你看，代码瞬间变得干净了。不用写if判断，不用手动raise异常。一行搞定。

2. assert的优缺点分析

任何工具都有两面性。assert也不例外。咱们客观聊聊。

优点

• 代码简洁：一行顶三行，可读性高
• 语义明确：看到assert就知道「这里是个不变条件」
• 零开销：生产环境可以全局关闭，不影响性能
• 调试友好：出错时直接定位到问题行，堆栈清晰

缺点

• 可被全局禁用：用-O或-OO启动Python时，所有assert都不执行
• 错误信息有限：只能带一个字符串，没法做复杂处理
• 不适合做正式校验：比如用户输入验证、数据完整性检查
• 不能捕获：AssertionError如果没被except捕获，程序直接崩溃

3. 适用场景 vs 不适用场景

到底什么时候该用assert？我总结了一张表：

说白了，assert最适合的场景是：开发阶段帮你抓bug，但生产环境关掉也无所谓。

4. 实战中的最佳实践

我在多个项目里总结了几条经验，分享给你：

1. assert只检查「不应该发生」的情况，而不是「可能发生」的情况。比如用户输入为空是「可能发生」的，应该用if处理；而列表索引越界是「不应该发生」的，可以用assert。
2. 错误信息要写清楚。别写assert x > 0就完事了。写成assert x > 0, f"x的值是{x}，预期大于0"，调试时能省你半小时。
3. 不要用assert做流程控制。比如assert func() == True然后继续执行——如果assert被关了，func()都不会被调用，逻辑就全乱了。
4. 团队约定好assert的使用规范。我习惯在项目文档里写清楚：哪些模块可以用assert，哪些必须用if+raise。避免新人乱用。

5. 知识体系总览

下面这张图帮你理清本章的核心逻辑：

6. 一个完整的实战例子

最后，给你看一段我在真实项目中写过的代码。这是一个二分查找的实现，里面用了assert做不变条件检查：

def binary_search(arr, target):
    """在有序数组中查找目标值，返回索引，未找到返回-1"""
    left, right = 0, len(arr) - 1
    
    # 断言：数组必须有序
    assert all(arr[i] <= arr[i+1] for i in range(len(arr)-1)), "数组必须升序排列"
    
    while left <= right:
        # 断言：区间必须有效
        assert 0 <= left <= right < len(arr), f"区间异常: left={left}, right={right}"
        
        mid = (left + right) // 2
        if arr[mid] == target:
            return mid
        elif arr[mid] < target:
            left = mid + 1
        else:
            right = mid - 1
    
    # 断言：如果执行到这里，target一定不在数组中
    assert target not in arr, f"逻辑错误：target={target}应该在数组中"
    return -1

你看，这些assert就像一个个「哨兵」。开发调试时，它们帮我快速发现逻辑漏洞。上线时，用-O关掉，性能零损耗。两全其美。

好了，关于assert就聊这么多。记住一句话：assert是你的debug好帮手，但不是你的安。用对地方，事半功倍；用错地方，后患无穷。