pytest中yield fixture的清理逻辑写在yield之后，与yield同缩进且仅出现一次，确保测试结束（含异常或中断）后自动执行；setup失败则yield不执行，teardown不触发。

pytest 中 yield fixture 的清理逻辑怎么写？

pytest 的 yield 用在 fixture 里，不是为了“生成值”，而是划出一个分界线：yield 前是 setup，yield 后是 teardown。pytest 会在测试执行完后自动运行 yield 后的代码，哪怕测试抛异常也照常执行。

• yield 后的代码必须和 yield 在同一个缩进层级，且只能出现一次
• yield 可以带返回值（供测试函数使用），但不带也完全合法
• 如果 fixture 被多个测试复用，每次测试结束后都会独立触发 yield 后的清理逻辑

@pytest.fixture
def temp_file():
    path = "/tmp/test_data.json"
    with open(path, "w") as f:
        f.write('{"ok": true}')
    yield path  # 测试拿到这个 path
    os.remove(path)  # 测试结束必执行，哪怕 test 报错

为什么不用 try/finally？yield 更可靠

try/finally 看似等价，但 yield fixture 由 pytest 统一管理生命周期，能保证：

• 即使测试进程被 SIGINT 中断（比如 Ctrl+C），pytest 仍会尝试执行 yield 后逻辑
• 多个 fixture 嵌套时，pytest 按依赖顺序逆序执行 teardown，不会错乱
• 与 scope="module" 或 scope="session" 配合时，teardown 时机明确（模块结束 / 会话结束）

常见错误：把清理逻辑写在 yield 后，但又在 yield 前抛了异常（比如文件创建失败），这时 yield 根本没执行，teardown 就不会触发。应确保 setup 部分尽量轻量，或提前校验前提条件。

yield fixture 和 return fixture 的行为差异

• return fixture：只做 setup，没有自动 teardown；想清理就得靠测试函数自己调（不可靠）
• yield fixture：天然支持“成对”资源管理，类似 contextlib.contextmanager

典型误用场景：

librosa Python 音频与音乐分析库

librosa 是一个用于音频和音乐分析的 Python 库。它提供了特征提取、频谱分析、节拍跟踪、起始检测和音频可视化的工具，基于 NumPy 和 SciPy 构建，用于科学音频计算。

下载

• 数据库连接：用 yield conn，后面 conn.close()
• 临时目录：用 yield tmpdir，后面 shutil.rmtree(tmpdir)
• 启动子进程：用 yield proc，后面 proc.terminate(); proc.wait()

注意：yield 后不能有 return，否则 pytest 会报 FixtureFinalizerNotCalledWarning。

清理失败时 pytest 会沉默吗？

不会。yield 后的代码如果抛异常，pytest 默认把它当作测试失败的一部分，并显示在结果里，但不影响测试本身的 pass/fail 判定——也就是说，测试函数通过了，但 cleanup 报错，整体结果仍是 failed。

• 这是故意设计：资源泄漏比单次测试失败更危险
• 若 cleanup 异常可容忍（比如删不存在的文件），需显式捕获：

yield db
try:
    db.drop_all()
except Exception:
    pass  # 或 log.warning

复杂点在于：多个 fixture 存在依赖时，一个 cleanup 失败可能导致后续 fixture 的 teardown 被跳过。务必让每个 fixture 的 teardown 尽可能幂等、无副作用。

Python免费学习笔记（深入）：立即使用
在学习笔记中，你将探索 Python 的核心概念和高级技巧！