Question

我有一个使用@ pytest.mark.parametrize参数化的测试，以便可以使用不同的参数执行相同的测试功能。 @ pytest.mark.parametrize使用定义为函数的ids参数，以便为每个参数生成自定义测试ID。我的测试功能将一个文件写入磁盘，该文件将给定参数的实际结果与预期结果之间的差异制成表格。我想在文件中报告测试ID。有没有我可以调用的API，该API可以告诉我特定的测试执行适用什么测试ID？

Answer 1

您可以使用 Request Fixture 中的 request.node.callspec.id。

但请注意，该值取决于 ids 函数“生成字符串表示以构成测试 ID 的一部分”的方式。根据 Different options for test IDs 上的 pytest 文档，参数的字符串化版本通常会添加到测试 ID 中。

例如，如果 ids 函数生成的 ID 不包含字符串化参数或不依赖任何其他内容：

import random
def generate_random_id(param) -> str:
    return f'{random.randint(1,100)}'

testdata = ['argA', 'argB']

@pytest.mark.parametrize('param', testdata, ids=generate_random_id)
def test_foo(param, request):
    print(request.node.callspec.id)

然后您将按原样获得生成的 ID：

tests/test_a.py::test_foo[78] PASSED      
tests/test_a.py::test_foo[10] PASSED      

============================ PASSES =============================
_________________________ test_foo[78] __________________________
--------------------- Captured stdout call ----------------------
78
_________________________ test_foo[10] __________________________
--------------------- Captured stdout call ----------------------
10

如果 ids 函数包含参数的字符串表示，那么您需要某种分隔符来手动拆分参数和测试 ID：

import random
def generate_random_id(param):
    test_id = random.randint(1, 100)
    return f'{param}-{test_id}'  

testdata = ['argA', 'argB']

@pytest.mark.parametrize('param', testdata, ids=generate_random_id)
def test_foo(param, request):
    param, test_id = request.node.callspec.id.split('-')
    print(param)
    print(test_id)

tests/test_a.py::test_foo[argA-43] PASSED 
tests/test_a.py::test_foo[argB-37] PASSED 

============================ PASSES =============================
_______________________ test_foo[argA-43] _______________________
--------------------- Captured stdout call ----------------------
argA
43
_______________________ test_foo[argB-37] _______________________
--------------------- Captured stdout call ----------------------
argB
37

如果有多个参数，那么它会变得更加复杂，因为所有参数 + ids 组合都连接在一起以创建完整的测试 ID。您将需要一个不同于 -:

的单独分隔符

import random
def generate_random_id(param):
    test_id = random.randint(1, 100)
    return f'{param}+{test_id}'  # Use '+' to separate param+ID

testdata = [
    ('argA', 'argB'),
    ('argC', 'argD'),
]

@pytest.mark.parametrize('param1, param2', testdata, ids=generate_random_id)
def test_foo(param1, param2, request):
    print(request.node.callspec.id)
    params = request.node.callspec.id.split('-')
    for a_param in params:
        param, test_id = a_param.split('+')
        print(param)
        print(test_id)

tests/test_a.py::test_foo[argA+70-argB+25] PASSED         
tests/test_a.py::test_foo[argC+97-argD+30] PASSED          

===================================== PASSES =====================================
___________________________ test_foo[argA+70-argB+25] ____________________________
------------------------------ Captured stdout call ------------------------------
argA+70-argB+25
argA
70
argB
25
___________________________ test_foo[argC+97-argD+30] ____________________________
------------------------------ Captured stdout call ------------------------------
argC+97-argD+30
argC
97
argD
30

Answer 2

您可以使用：

os.environ.get('PYTEST_CURRENT_TEST')

获取当前参数化执行。

已经有人问过类似的问题，并且还有很多其他选项：py.test: how to get the current test's name from the setup method?

如果您正在寻找一种更动态的方法，请查看：https://hackebrot.github.io/pytest-tricks/param_id_func/，这是一个 pytest 插件，可以减轻痛苦。

在执行测试期间查找测试ID

2 个答案: