参数化测试也取决于 pytest 中的参数化值

我有一个测试，我有一个设置方法，应该收到一个dataset和一个测试函数，应该为每个运行data in dataset

基本上我需要类似的东西：

datasetA = [data1_a, data2_a, data3_a]
datasetB = [data1_b, data2_b, data3_b]

@pytest.fixture(autouse=True, scope="module", params=[datasetA, datasetB])
def setup(dataset):
    #do setup
    yield
    #finalize

#dataset should be the same instantiated for the setup
@pytest.mark.parametrize('data', [data for data in dataset]) 
def test_data(data):
    #do test

它应该像这样运行：

• 设置（数据集A）
• 测试（数据1_a）
• 测试（数据2_a）
• 测试（data3_a）
• 设置（数据集B）
• 测试（数据1_b）
• 测试（数据2_b）
• 测试（data3_b）

然而，似乎不可能像我在示例中那样对夹具获得的变量进行参数化。

我可以让我的函数使用固定装置并在测试方法内进行迭代：

def test_data(dataset):
    for data in dataset:
        #do test

但随后我会进行一次大型测试，而不是针对每种情况进行单独的测试，这是我不希望进行的。

有什么方法可以实现这一点吗？

Thanks!

答案#1：如果严格遵循测试设计，那么它应该如下所示：

import pytest

datasetA = [10, 20, 30]
datasetB = [100, 200, 300]

@pytest.fixture
def dataset(request):
    #do setup
    items = request.param
    yield items
    #finalize

@pytest.fixture
def item(request, dataset):
    index = request.param
    yield dataset[index]

#dataset should be the same instantiated for the setup
@pytest.mark.parametrize('dataset', [datasetA, datasetB], indirect=True)
@pytest.mark.parametrize('item', [0, 1, 2], indirect=True)
def test_data(dataset, item):
    print(item)
    #do test

注意两者的间接参数化item & dataset。参数值将被传递到同名的装置request.param。在本例中，我们使用索引的假设是数据集的长度相同，均为 3 个项目。

下面是它的执行方式：

$ pytest -s -v -ra test_me.py 
test_me.py::test_data[0-dataset0] 10
PASSED
test_me.py::test_data[0-dataset1] 100
PASSED
test_me.py::test_data[1-dataset0] 20
PASSED
test_me.py::test_data[1-dataset1] 200
PASSED
test_me.py::test_data[2-dataset0] 30
PASSED
test_me.py::test_data[2-dataset1] 300
PASSED