Pytest 的底层源码最应该学习的技术

在 Pytest 的底层源码中它并不是简单地“在指定函数地方 Mock”也不是单纯地“增加异常捕获”。它的设计要精妙得多1. 不是“在指定地方 Mock”而是“在内存中偷梁换柱”你理解的“在指定的函数地方 mock”听起来像是在业务代码里写 if is_test: mock() else: run()。但 Pytest 绝对不碰你的业务代码。它的底层原理是**“猴子补丁Monkey Patching”**在测试执行前Pytest 直接去内存的模块字典dict里把真实函数的内存地址强制替换成了假函数的内存地址。效果你的业务代码浑然不觉它以为自己调用的还是真实的数据库或网络请求但实际上它调用的是 Pytest 塞进去的假函数。测试结束后Pytest 再把真实地址换回来。2. 不是简单的“增加异常捕获”而是“生命周期拦截”Pytest 并没有在你的业务函数外面套一个 try…except。它利用了我们之前聊过的钩子机制Hook和Fixture 机制。Pytest 在底层有一个严格的执行协议pytest_runtest_protocol它把测试分成了三个阶段Setup准备阶段在这里注入入参执行内存替换Mock。Call执行阶段在这里运行你的业务函数。Teardown清理阶段在这里还原内存清理环境。如果业务函数抛出了异常Pytest 的底层框架会接住这个异常并把它转化为一个“测试失败FAILED”的状态而不是让程序直接崩溃。3. 不是普通的“看返回结果”而是“AST 断言重写”你平时写 assert result expected在 Pytest 底层这行代码在运行前就被语法树AST重写了。Pytest 把它变成了一段极其复杂的代码不仅判断结果对不对还会把 result 和 expected 的每一个属性、每一个变量的值都打印出来。这就是为什么 Pytest 报错时能给你提供那么丰富、那么详细的上下文信息。