如何在测试类setup_class及非测试函数中使用pytest fixture？

问题场景 #

conftest.py中定义的Session作用域Fixture #

# conftest.py
import pytest
import random
import string


@pytest.fixture(scope = "session")
def run_id():
    random_id = ''.join(random.choices(string.ascii_letters + string.digits, k=16))
    _run_id = "fail_fast_" + random_id
    return _run_id

测试文件test.py中的代码 #

# test.py

import pytest
import os


def get_run_id(run_id):
    return run_id


class TestTrainingAndValidationPipeline:
    
    @classmethod
    def setup_class(cls):
        cls._run_id = get_run_id() # 报错："TypeError: get_run_id() missing 1 required 
                                   #          positional argument: 'run_id'"

        print(f"Setting up {cls._run_id}")
    
    @classmethod
    def teardown_class(cls):
       print(f"Tearing down {cls._run_id}")

    def test_attribute_change(self):
        print("hello")
        self._run_id= "world"
        print(self._run_id)

技术问询 #

能否在setup_class这类非测试函数中使用pytest fixture设置测试类的类变量？若无法实现，如何通过get_run_id函数获取run_id fixture的值？

解决方案 #

1. 无法直接在setup_class中使用Fixture #

pytest的setup_class是unittest风格的类级前置方法，它不属于pytest的Fixture注入体系，无法直接接收Fixture参数，这也是当前报错的核心原因。

2. 最优替代：用pytest类级Fixture替代setup_class #

可以用@pytest.fixture(scope="class")替换原生的setup_class，这样就能直接注入run_id fixture，同时完成类变量的设置，还能整合teardown逻辑：

修改后的test.py代码：

# test.py
import pytest
import os


def get_run_id(run_id):
    return run_id


class TestTrainingAndValidationPipeline:
    # 用类级Fixture替代setup_class，autouse=True确保自动执行
    @pytest.fixture(scope="class", autouse=True)
    def setup_class_with_fixture(self, run_id):
        # 将Fixture值通过get_run_id处理后设为类变量
        self.__class__._run_id = get_run_id(run_id)
        print(f"Setting up {self.__class__._run_id}")
        # yield后编写teardown逻辑，替代原teardown_class
        yield
        print(f"Tearing down {self.__class__._run_id}")

    def test_attribute_change(self):
        print("hello")
        self._run_id = "world"
        print(self._run_id)

• autouse=True让Fixture自动在类执行前运行，和setup_class效果一致
• Fixture方法可直接接收run_id参数，完成注入后调用get_run_id处理值并设置为类变量
• yield后面的代码会在所有测试用例执行完后运行，替代原teardown_class的功能

3. 兼容原有setup_class的方案 #

如果一定要保留setup_class结构，可以在第一个测试方法中注入run_id，再传递给类变量：

# test.py
import pytest
import os


def get_run_id(run_id):
    return run_id


class TestTrainingAndValidationPipeline:
    _run_id = None

    @classmethod
    def setup_class(cls):
        # 此处无法直接获取Fixture，留空后续由测试方法填充
        pass
    
    @classmethod
    def teardown_class(cls):
       print(f"Tearing down {cls._run_id}")

    def test_init_run_id(self, run_id):
        # 第一个执行的测试用例注入run_id，设置为类变量
        if not self.__class__._run_id:
            self.__class__._run_id = get_run_id(run_id)
            print(f"Setting up {self.__class__._run_id}")

    def test_attribute_change(self):
        print("hello")
        self._run_id = "world"
        print(self._run_id)

这种方法需要确保test_init_run_id是类中第一个执行的测试用例，这样后续测试和teardown_class就能正常使用类变量_run_id。

内容的提问来源于stack exchange，提问作者CutePoison