如何将pytest fixture输入传递给pytest.mark.parametrize()？

这个报错的核心原因很明确：pytest的参数化（@pytest.mark.parametrize）是在测试收集阶段执行的，而你定义的input_sequence fixture是在测试运行阶段才会被实例化的。在收集阶段，input_sequence这个变量根本还不存在，所以直接在parametrize里引用range(len(input_sequence))肯定会报NameError。

下面给你一个简单直接的解决方案，用pytest的pytest_generate_tests钩子函数来动态生成测试参数：

步骤1：修改测试文件 #

移除测试函数上的@pytest.mark.parametrize装饰器，然后添加pytest_generate_tests钩子来动态生成rotation参数：

import pytest
from typing import List

def pytest_generate_tests(metafunc):
    # 检查当前测试函数是否需要rotation参数，避免影响其他测试
    if "rotation" in metafunc.fixturenames:
        # 直接从配置中获取命令行选项的值，和你的fixture逻辑一致
        csv_str = metafunc.config.getoption("--sequence-csv")
        input_sequence = [float(num) for num in csv_str.split(",")]
        # 生成所有可能的rotation值并传递给测试函数
        metafunc.parametrize("rotation", range(len(input_sequence)))

def test_sequence_rotation(
    input_sequence: List[float],
    rotation: int,
) -> None:
    sequence = input_sequence[rotation:] + input_sequence[:rotation]
    print(f"Testing sequence {sequence}")
    # 这里可以添加你的断言逻辑

步骤2：保持conftest.py不变 #

你原来定义的input_sequence fixture和命令行选项逻辑不需要修改，继续保留即可：

import pytest
from typing import List
from _pytest.config.argparsing import Parser
from _pytest.fixtures import SubRequest

def pytest_addoption(parser: Parser) -> None:
    parser.addoption("--sequence-csv", type=str, required=True)

@pytest.fixture()
def input_sequence(request: SubRequest) -> List[float]:
    csv = request.config.getoption("--sequence-csv")
    return [float(i) for i in csv.split(",")]

为什么这个方案可行？ #

pytest_generate_tests是pytest专门提供的钩子函数，它在测试收集阶段运行，可以直接访问pytest的配置（包括命令行选项），这样我们就能在收集阶段生成rotation的参数列表，而不需要依赖还未实例化的fixture。

运行测试 #

和之前一样，用命令行传递序列参数即可：

pytest --sequence-csv "1.5,2.5,3.5,4.5"

这会自动生成4个测试用例，对应rotation值0、1、2、3。

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