Python pytest断言写法详解

Python pytest 的断言机制彻底摆脱了 unittest 的繁复束缚，直接拥抱原生 assert 语法——无需继承 TestCase、不用 self.assert* 方法，测试函数只需以 test_ 开头或 _test 结尾，失败时即自动展开变量值、高亮差异根源；其背后是 AST 编译期重写技术，让 assert response.status_code == 200 失败时清晰显示 “assert 404 == 200” 并溯源到 response.status_code 的实际值；但要注意避开链式调用、函数内联、布尔组合等展开盲区，善用赋值拆解、pytest.approx() 处理浮点、pytest.raises()/pytest.warns() 精准捕获异常与警告——掌握这些，你才能真正释放 pytest 断言简洁、智能又强大的生产力。

pytest 里直接用 assert 就行，不用写 self.assertEqual

pytest 的核心设计就是让断言回归 Python 原生语义——你写什么，它就按什么报错。不需要继承 TestCase，不出现 self.assert* 系列方法。只要在测试函数里写 assert，pytest 就会自动捕获、重写、展开并高亮失败原因。

常见错误现象：有人照搬 unittest 写法，在 pytest 测试函数里写 self.assertEqual(a, b)，结果报 AttributeError: 'TestSomething' object has no attribute 'assertEqual' ——因为没继承、也没 self 上下文。

• 测试函数必须是普通函数，以 test_ 开头或以 _test 结尾
• 不要加 class TestXXX(unittest.TestCase): 这类结构（除非你明确想混用 unittest）
• assert 后面跟任意合法 Python 表达式，比如 assert len(items) == 3、assert "error" not in log_output

assert 失败时为什么能看清具体值？

因为 pytest 在运行时重写了 assert 语句的 AST，把原始表达式拆解成变量名 + 实际值 + 比较逻辑。不是靠异常消息拼接，而是编译期介入。

使用场景：当你写 assert response.status_code == 200 却失败了，pytest 不会只报 “AssertionError”，而是显示类似：

assert 404 == 200
 +  where 404 = response.status_code

这依赖于 pytest 的 assertion rewriting 机制，默认开启；如果测试文件被动态导入或用 exec() 执行，可能失效。

• 确保测试文件后缀是 .py，且由 pytest 直接发现执行（如 pytest test_api.py）
• 禁用重写的配置项是 --assert=plain，一般别碰
• 自定义 __repr__ 或 __str__ 会影响展示效果，但不影响断言逻辑

哪些 assert 写法容易踩坑？

不是所有 assert 表达式都能被清晰展开。pytest 对复杂链式调用、函数调用、布尔运算的支持有边界。

常见错误现象：写 assert get_user().name == "alice" 失败时只报 AssertionError，没有中间值提示；或者 assert a and b 失败后看不出哪个子表达式为假。

• 避免在 assert 右侧直接调用函数，先赋值再断言：user = get_user(); assert user.name == "alice"
• assert x in y 能展开，但 assert not x in y 展开效果弱，建议改写为 assert x not in y
• 多个条件用逗号拼一起（assert a == 1, b == 2）是语法错误，正确是用 and 连接
• 浮点数比较慎用 ==，优先用 pytest.approx()： assert result == pytest.approx(3.14, abs=1e-3)

需要更细粒度控制时，用 pytest.raises 和 pytest.warns

原生 assert 无法检查“是否抛出异常”或“是否发出警告”。这时候必须换专用上下文管理器。

使用场景：验证某段代码必须报 ValueError，或必须触发 DeprecationWarning。

• 写 assert raise ValueError 是无效语法，会直接报错
• 正确写法是：with pytest.raises(ValueError, match="invalid.*input"): + 缩进执行代码
• match 参数支持正则，不填也能捕获，但推荐加上便于定位
• 警告检测同理：with pytest.warns(DeprecationWarning):

这些上下文管理器本身不依赖 assertion rewriting，但它们内部仍会配合 pytest 的异常捕获机制做精准匹配。最容易被忽略的是：忘记缩进执行语句，导致上下文退出后才执行，断言永远不生效。

理论要掌握，实操不能落！以上关于《Python pytest断言写法详解》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！