3

我有一个场景，我有一个参数列表。每个参数都可以生成另一个参数列表，我想根据这两个参数对我的测试进行参数化。

类似于我在for loop下面所做的事情，但使用了参数化。并且应该针对任何一组参数一起test_1执行test_2

有什么帮助吗？？

param1 = [<some_list>]
#param2 = [<some_list_Generated_based_on_Param1>]

for arg1 in param1:
    param2 = get_param2(arg1)
    for arg2 in param2
        def test_1():
            ...
            pass
        def test_2():
            ...
            pass
4

1 回答 1

1

我怀疑是否有直接的方法可以实现这一目标。但是通过扩展 pytest 插件pytest_generate_tests，我们可以做到。

这是我对另一个parametrize问题的回答。使用夹具返回值作为 mark.parametrize() 中的值

这是我的解决方案。

  1. 标记要进行两级参数化的测试。
  2. 获取要进行二级参数化的测试，并获取一级参数化。
  3. 基于第一级参数化，生成第二级。
  4. 分配的测试用例仅使用二级参数化或根据您的要求同时使用。

示例测试：

import pytest

@pytest.mark.two_level_parametrization
@pytest.mark.parametrize("first_level", [1, 2, 3])
def test_two_level_parametrization(second_level):
    print(second_level)

Conftest.py：

import pytest

def pytest_generate_tests(metafunc):
    if hasattr(metafunc.function, 'two_level_parametrization'):
        # You may need more marker or add if statement based on the arg of parametrize if you want to implement multiple two-level paremetrize
        first_parametrize = metafunc.function.parametrize
        metafunc.function.parametrize = [calculate_second_parametrize(first_parametrize)]

        # # If you want to keep first level parametrization:
        # metafunc.function.parametrize = [first_parametrize, calculate_second_parametrize(first_parametrize)]

def calculate_second_parametrize(first_parametrize):
    # Define your own method here to generate the parametrize object you need.

    input_params = first_parametrize.args[1]
    output_params = input_params + [-1 * num for num in input_params]
    return pytest.mark.parametrize('second_level', output_params)

输出：

test_01.py::test_two_level_parametrization[1] 1
PASSED
test_01.py::test_two_level_parametrization[2] 2
PASSED
test_01.py::test_two_level_parametrization[3] 3
PASSED
test_01.py::test_two_level_parametrization[-1] -1
PASSED
test_01.py::test_two_level_parametrization[-2] -2
PASSED
test_01.py::test_two_level_parametrization[-3] -3
PASSED
于 2016-06-25T18:36:58.613 回答