使用函数进行 Pytest 参数化测试

Question

我正在尝试使用产生参数值的函数对pytest测试进行参数化。我希望它为每个值单独运行测试。下面是示例代码：

def provider():
yield pytest.param([1, 2])

class Test:
@pytest.mark.parametrize("a", provider())
def test(self, a, logger):
logger.info("a: {}".format(a))
@pytest.mark.parametrize("a", [1, 2])
def test_2(self, a, logger):
logger.info("a: {}".format(a))

是否可以使用提供程序函数对测试进行参数化，使其像test_2一样工作。

以下是我从上述测试中获得的日志：

2019-12-09 14:39:36 test[a0] start
a: [1, 2]
2019-12-09 14:39:36 test[a0] passed
2019-12-09 14:39:36 test_2[1] start
a: 1
2019-12-09 14:39:36 test_2[1] passed
2019-12-09 14:39:36 test_2[2] start
a: 2
2019-12-09 14:39:36 test_2[2] passed

Answer 1

您的函数provider()生成单个参数。您必须遍历列表并生成每个项目。

def provider():
for param in [1, 2]:
yield pytest.param(param)

根据您的确切需求，您可以将其简化为：

def provider():
yield from [1, 2]

甚至只是：

def provider():
return [1, 2]