如何在pytest中进行依赖参数化

Question

例如，我有以下测试数据：

PARAMS = {'pic1': [1, 2, 3], 'pic2': [14, 15], 'pic3': [100, 200, 300]}

我需要从这个PARAMS下载每个关键图片，并生成单独的测试[1, 2, 3]，它将使用这个图片。之后，当'pic1': [1, 2, 3]对的每次测试结束时，删除图片，然后下载下一张，以此类推。。。粗略地说，生成的测试应该是这样的：

test_pic[pic1-1]
test_pic[pic1-2]
test_pic[pic1-3]
test_pic[pic2-14]
test_pic[pic2-15]
test_pic[pic3-100]
test_pic[pic3-200]
test_pic[pic3-300]

但内部将是下载图像逻辑。

我没有找到在pytest中如何做到这一点的方法。

请帮忙。

Answer 1

对我来说，这似乎是一项间接参数化的任务。

准备参数

首先：pytest.mark.parametrize希望参数以元组列表的形式传递，数据按参数名称排序，例如：

@pytest.mark.parametrize('spam,eggs', [(1, 2), (3, 4), (5, 6)])

将生成三个测试：

• CCD_ 5
• CCD_ 6
• spam=5, eggs=6

因此，您必须将PARAMSdict转换为元组列表，每个dict键有一个数字。有很多方法可以做到这一点，其中一个解决方案是：

PARAMS = {'pic1': [1, 2, 3],
'pic2': [14, 15],
'pic3': [100, 200, 300]}
test_pic_params = [(key, el) for key, nums in PARAMS.items()
for el in nums]

@pytest.mark.parametrize('file,num', test_pic_params)
def test_pic(file, num):
assert True

检查测试是否正确生成：

$ pytest --collect-only --quiet test_spam.py
test_spam.py::test_pic[pic1-1]
test_spam.py::test_pic[pic1-2]
test_spam.py::test_pic[pic1-3]
test_spam.py::test_pic[pic2-14]
test_spam.py::test_pic[pic2-15]
test_spam.py::test_pic[pic3-100]
test_spam.py::test_pic[pic3-200]
test_spam.py::test_pic[pic3-300]
no tests ran in 0.07 seconds

间接参数化

现在您希望在测试之前处理file参数，因此测试将获得pic1而不是pic1的下载文件。这可以通过间接参数化来实现。你需要做的是：

• 实现一个名为file的fixture(重要的是该fixture与测试参数同名，否则pytest将无法识别和应用它(
• 将CCD_ 14添加到参数化标记中

通过这种方式，首先将pic1传递给file()夹具，然后将夹具的结果传递给测试。扩展示例：

import pathlib
import pytest

PARAMS = {'pic1': [1, 2, 3], 
'pic2': [14, 15],
'pic3': [100, 200, 300]}
test_pic_params = [(key, el) for key, nums in PARAMS.items()
for el in nums]

@pytest.fixture
def file(request):
pic = request.param
# this will just create an empty file named 'pic1' etc;
# replace it with file download logic
filename = pathlib.Path(pic)
filename.touch()
# pass filename instead of pic to test
yield filename
# after test finishes, we remove downloaded file
filename.unlink()

@pytest.mark.parametrize('file,num', test_pic_params, indirect=['file'])
def test_pic(file, num):
assert file.is_file()

编辑：

我需要的是在测试test_pic[pic1-1] , test_pic[pic1-2], test_pic[pic1-3]之后删除文件'pic1'。然后下载新文件pic2。

尽管这肯定是可能的，但请记住，这将违反单个测试运行的原子性，因此，例如，您将失去并行运行测试的能力。

如果您想跟踪测试运行的当前状态，只需在fixture中进行即可。当号码是相应列表中的第一个时，下载该文件；删除最后一个号码上的文件：

@pytest.fixture
def file(request):
pic = request.param
num = request.node.callspec.params['num']
filename = pathlib.Path(pic)
if num == PARAMS[pic][0]:  # download here
filename.touch()
yield filename
if num == PARAMS[pic][-1]:  # remove here
filename.unlink()

海事组织更好的办法是使用磁盘缓存，在第一次下载时缓存文件；这将使测试再次成为原子运行。