Python Pytest夹具示例

fixture 是用于为测试提供预设环境或数据的函数，1. 使用 @pytest.fixture 装饰器定义 fixture；2. 在测试函数中以参数形式注入 fixture；3. yield 之前执行 setup，之后执行 teardown；4. 通过 scope 参数控制作用域，如 function、module 等；5. 将共用 fixture 放在 conftest.py 中实现跨文件共享，从而提升测试的可维护性和复用性。

在使用 pytest 进行测试时，fixture 是一个非常强大的功能，用于为测试函数提供预设的测试环境或数据。下面是一个简单但完整的 pytest fixture 示例，帮助你理解如何定义和使用 fixture。

✅ 基本 fixture 示例：数据库连接模拟

假设我们要测试一个操作“用户数据”的模块，需要一个数据库连接对象。

# test_user.py

import pytest

# 模拟数据库
class Database:
    def __init__(self):
        self.users = {}

    def add_user(self, user_id, name):
        self.users[user_id] = name

    def get_user(self, user_id):
        return self.users.get(user_id)

    def clear(self):
        self.users.clear()

# 定义 fixture：提供一个干净的数据库实例
@pytest.fixture
def db():
    print("\n[Setup] 创建数据库实例")  # 可以看到执行顺序
    database = Database()
    yield database  # 返回给测试函数使用
    print("[Teardown] 清理数据库")
    database.clear()

# 测试用例使用 fixture
def test_add_user(db):
    db.add_user(1, "Alice")
    assert db.get_user(1) == "Alice"

def test_get_user_not_exists(db):
    assert db.get_user(99) is None

def test_add_multiple_users(db):
    db.add_user(1, "Alice")
    db.add_user(2, "Bob")
    assert len(db.users) == 2
    assert db.get_user(2) == "Bob"

? 运行测试

确保你已经安装 pytest：

pip install pytest

运行测试：

pytest test_user.py -v

输出类似：

test_user.py::test_add_user[Setup] 创建数据库实例
PASSED[Teardown] 清理数据库

test_user.py::test_get_user_not_exists[Setup] 创建数据库实例
PASSED[Teardown] 清理数据库

test_user.py::test_add_multiple_users[Setup] 创建数据库实例
PASSED[Teardown] 清理数据库

每个测试函数都会获得一个独立的 db 实例，因为 fixture 函数在每次测试前重新执行。

? fixture 作用域（scope）

你可以控制 fixture 的复用范围，比如让多个测试共享同一个实例：

@pytest.fixture(scope="module")
def db():
    print("\n[Module Setup] 创建数据库一次")
    database = Database()
    yield database
    print("[Module Teardown] 清理数据库一次")
    database.clear()

• scope="function"：默认，每个测试函数调用一次（最常见）
• scope="class"：每个测试类调用一次
• scope="module"：每个 .py 文件调用一次
• scope="session"：整个测试会话只执行一次

? conftest.py：共享 fixture

如果你有多个测试文件需要使用同一个 fixture，可以把它放在 conftest.py 中：

# conftest.py

import pytest

@pytest.fixture
def db():
    database = Database()
    yield database
    database.clear()

这样，所有同目录或子目录下的测试文件都可以直接使用 db，无需导入。

✅ 总结关键点

• @pytest.fixture 装饰器定义可重用的测试资源
• yield 之前是 setup，之后是 teardown
• fixture 函数名作为参数传入测试函数，自动注入
• 使用 conftest.py 实现跨文件共享 fixture
• 可通过 scope 控制生命周期，提升性能

基本上就这些。fixture 让测试更干净、可复用、易维护。

以上是Python Pytest夹具示例的详细内容。更多信息请关注PHP中文网其他相关文章！