如何让pytest中第二个参数化变量成为第一个的函数？

你遇到的问题本质是：pytest的@pytest.mark.parametrize装饰器是在函数定义阶段执行的，第二个参数化装饰器运行时，第一个参数化的type值还没被传入测试用例，自然没法用它来计算VAR_THAT_IS_FUNCTION_OF_TYPE。

下面给两种实用的解决方法：

方法一：用pytest_generate_tests钩子动态生成用例 #

这是最直接的方案，在测试收集阶段就根据type的所有可能值，动态生成对应的val参数组合。

示例代码：

import pytest

# 替换成你实际根据type计算上限的逻辑
def get_upper_limit(type_str):
    type_to_limit = {'a': 2, 'b': 5, 'c': 3}
    return type_to_limit[type_str]

def pytest_generate_tests(metafunc):
    # 只处理包含type和val参数的测试函数
    if 'type' in metafunc.fixturenames and 'val' in metafunc.fixturenames:
        all_test_cases = []
        # 遍历所有type参数值
        for type_val in ['a', 'b', 'c']:
            upper = get_upper_limit(type_val)
            # 生成当前type对应的所有val值组合
            for val in range(0, upper):
                all_test_cases.append((type_val, val))
        # 给测试函数参数化
        metafunc.parametrize('type, val', all_test_cases)

def test_example(type, val):
    # 你的测试逻辑
    assert val < get_upper_limit(type)

方法二：间接参数化+Fixture #

如果只想针对单个测试用例处理，可以结合indirect=True和Fixture，让Fixture在测试运行时获取当前的type参数，生成对应的val范围。

示例代码：

import pytest

def get_upper_limit(type_str):
    type_to_limit = {'a': 2, 'b': 5, 'c': 3}
    return type_to_limit[type_str]

@pytest.fixture
def val(request):
    # 从当前测试用例的参数中获取type值
    current_type = request.node.callspec.params['type']
    upper = get_upper_limit(current_type)
    # 返回val的所有可能值，pytest会自动展开成多个测试用例
    return list(range(0, upper))

@pytest.mark.parametrize('type', ['a', 'b', 'c'])
# indirect=True表示让pytest调用名为val的Fixture来生成参数
@pytest.mark.parametrize('val', [[]], indirect=True)
def test_example(type, val):
    assert val < get_upper_limit(type)

两种方法都能解决你的问题，钩子函数适合全局统一处理多个测试用例，间接参数化更适合单个测试用例的场景。

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