それを達成する直接的な方法があるかどうかは疑問です。しかし、 pytest plugin を拡張するpytest_generate_tests
ことで、それを行うことができます。
これは別のparametrize
質問に対する私の答えです。mark.parametrize() の値としてフィクスチャの戻り値を使用する
これが私の解決策です。
- 2 レベルのパラメーター化を実行するテストをマークします。
- 2 レベルのパラメーター化を行うテストを取得し、最初のレベルのパラメーター化を取得します。
- 第 1 レベルのパラメーター化に基づいて、第 2 レベルを生成します。
- 割り当てられたテスト ケースは、要件に応じて、第 2 レベルのパラメーター化のみを使用するか、両方を使用します。
テスト例:
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