Question

正在测试的算法的输入是具有相关年份的字符串列表（事先已知），例如以下所有内容都是有效的输入：

['A_2018', 'B_2019', 'C_2018']
[]
['A_2018']

我有以下简单的年份装置：

@pytest.fixture(params=range(2018, 2025))
def year(request):
    return request.param

如果我为每个有效字符串创建单独的灯具：

@pytest.fixture(params=['A', ''])
def A(request, year):
    return request.param, year

@pytest.fixture(params=['B', ''])
def B(request, year):
    return request.param, year

等。并在以下测试中使用它们：

def test_foo(A, B):
    args = []
    if A[0]:
        args.append('%s_%d' % A)
    if B[0]:
        args.append('%s_%d' % B)
    assert old_algorithm(args) == new_algorithm(args)

我得到了

['A_2018', 'B_2018']
['A_2019', 'B_2019']
['A_2020', 'B_2020']

等。两个参数的年份总是相同的。

有没有办法创建所有组合？

Answer 1

夹具产生组合是一项要求吗？因为否则，为每个测试输入arg分别应用pytest.mark.parametrize会生成输入args组合。在下面的示例中，灯具A和B分别进行参数化，总共生成(2025 - 2018)**2次测试：

@pytest.fixture
def A(request):
    return 'A', request.param


@pytest.fixture
def B(request):
    return 'B', request.param


@pytest.mark.parametrize('A', range(2018, 2025), indirect=True, ids=lambda year: 'A({})'.format(year))
@pytest.mark.parametrize('B', range(2018, 2025), indirect=True, ids=lambda year: 'B({})'.format(year))
def test_foo(A, B):
    assert A[0] == 'A' and B[0] == 'B'

结果，产生了49个测试：

$ pytest --collect-only | grep collected
collected 49 items

我可以创建一个返回参数组合的pytest.fixture吗？

1 个答案: