在 Python 中使用 `pytest` 的夹具（fixture）时，通常夹具用于测试函数以提供初始化和清理操作。然而，在某些情况下，也可以在普通函数中使用夹具，但需要注意，夹具本质上是为测试函数设计的，因此在普通函数中使用时需要通过调用测试函数或者重构代码来实现。 ### 在普通函数中使用夹具 虽然 `pytest` 的夹具主要用于测试函数，但可以通过以下方式在普通函数中使用夹具： 1. **将夹具作为参数传递给普通函数** 可以通过在测试函数中调用夹具，并将夹具返回的值作为参数传递给普通函数。这种方式适用于需要在普通函数中使用夹具提供的数据或环境。 ```python import pytest @pytest.fixture def sample_data(): return [1, 2, 3, 4] def process_data(data): return sum(data) def test_process_data(sample_data): result = process_data(sample_data) assert result == 10 ``` 2. **直接在普通函数中调用夹具函数** 如果夹具函数是普通函数（没有使用 `@pytest.fixture` 装饰器），则可以直接调用它。但如果夹具已经被 `@pytest.fixture` 装饰，则不能直接调用，因为 `pytest` 会管理其生命周期。 ```python import pytest @pytest.fixture def setup_environment(): # 初始化环境 return {"config": "test"} # 普通函数中使用夹具 def use_fixture_in_function(): env = setup_environment() print(env) ``` 注意：直接调用夹具函数在 `pytest` 中是不推荐的，因为夹具的生命周期由 `pytest` 管理。更推荐的方式是通过测试函数间接使用夹具。 3. **使用 `request` 对象获取夹具** 在某些高级用例中，可以通过 `request` 对象获取夹具。例如，在模块级别或类级别定义的函数中使用夹具。 ```python import pytest @pytest.fixture def sample_config(): return {"timeout": 10} def get_fixture_value(request, fixture_name): return request.getfixturevalue(fixture_name) def test_use_fixture(request): config = get_fixture_value(request, "sample_config") assert config["timeout"] == 10 ``` ### 适用场景 - **数据共享**：当多个测试函数需要共享相同的测试数据时，可以使用夹具提供数据，并通过函数参数传递给普通函数。 - **环境初始化**：在测试前初始化数据库连接、文件系统等资源，并在普通函数中使用这些资源进行操作。 - **清理操作**：夹具可以定义 `yield` 语句用于执行清理操作，确保测试后资源被正确释放[^2]。 ### 注意事项 - **生命周期管理**：夹具的生命周期由 `pytest` 管理，普通函数中直接调用夹具可能会导致预期之外的行为。 - **模块化设计**：建议将夹具和普通函数的逻辑解耦，避免直接依赖夹具，提高代码的可维护性。 - **作用域**：夹具可以定义为 `function`、`class`、`module` 或 `session` 作用域，根据需求选择合适的作用域设置[^1]。 ---