Python pytest怎么实现类似Unittest的setup效果_使用yield实现后置处理
pytest 中 setup_method 和 setup_class 怎么用
pytest 本身不叫 setup,但行为上完全能替代 unittest 的 setUp 和 setUpClass。直接在测试类里定义 setup_method(每条测试前调用)或 setup_class(整个类首次运行前调用)就行,pytest 会自动识别并执行。
注意:这两个方法名必须拼写准确,且不能带参数(除了 self),否则 pytest 会忽略它们。
-
setup_method接收一个参数method,可用于判断当前运行的是哪个测试函数(比如跳过某些耗时操作) -
setup_class只在类内第一个测试开始前执行一次,适合初始化数据库连接、启动 mock server 等重操作 - 它们没有返回值要求,也不支持 yield;想做后置清理,得换别的机制
用 yield fixture 实现 setup + teardown 组合
真正灵活、推荐的方式是用带 yield 的 fixture —— 它天然支持“先 setup,再运行测试,最后 teardown”三段式流程。pytest 在 yield 前的部分当 setup,在 yield 后的部分当 teardown。
示例:
@pytest.fixture
def db_connection():
conn = create_test_db() # setup 阶段
yield conn # 测试用这个 conn
conn.close() # teardown 阶段,一定会执行(哪怕测试崩溃)
- 这个 fixture 默认作用域是
function,即每个测试函数独享一份 setup/teardown - 加
scope="class"就等效于setUpClass+tearDownClass - yield 后的代码即使测试抛异常也会执行,比 try/finally 更可靠
fixture scope 和 autouse 的坑
很多人误以为加了 autouse=True 就等于全局 setup,结果发现 teardown 没触发、或者被多次执行。关键点在于 scope 和 autouse 的组合逻辑:
Metaphor
一个新出炉的AI搜索引擎Metaphor
下载
-
autouse=True+scope="function":每个测试前 setup,测试后 teardown,安全可控 -
autouse=True+scope="session":整个 pytest 运行只 setup 一次,但 teardown 只在最后执行一次 —— 如果中间某个测试崩溃导致进程退出,teardown 可能根本不会跑 - 多个 autouse fixture 之间无执行顺序保证,别依赖它们的先后关系做资源依赖
更稳妥的做法是显式声明依赖,比如让 db_connection fixture 依赖 tmpdir,pytest 会自动按需调用。
yield fixture 被中断时的 teardown 行为
yield fixture 的 teardown 部分不是“绝对保险”的。以下情况它可能不执行:
- 测试进程中被
os._exit()强制终止(绕过 Python 正常退出流程) - pytest 进程被
SIGKILL(kill -9)杀死 - 测试中调用了
sys.exit()且没被 pytest 的异常捕获机制兜住(少见,但存在)
所以,涉及文件写入、端口绑定、临时进程启动这类操作,建议在 teardown 里加日志或状态检查,而不是假设“只要 yield 写了就一定执行”。