параметризованный тест с декартовым произведением аргументов в pytest

Ответ 1

Я могу придумать два способа сделать это. Один использует параметризованные светильники, и один параметризует тестовую функцию. Это зависит от вас, который вы найдете более элегантным.

Здесь параметризована тестовая функция:

import itertools
import pytest

numbers = [1,2,3,4,5]
vowels = ['a','e','i','o','u']
consonants = ['x','y','z']


@pytest.mark.parametrize('number,vowel,consonant',
    itertools.product(numbers, vowels, consonants)
)
def test(number, vowel, consonant):
    pass

Следует отметить, что второй аргумент декоратора параметриза может быть итерируемым, а не просто списком.

Вот как вы это делаете, параметризуя каждый прибор:

import pytest

numbers = [1,2,3,4,5]
vowels = ['a','e','i','o','u']
consonants = ['x','y','z']


@pytest.fixture(params=numbers)
def number(request):
    return request.param

@pytest.fixture(params=vowels)
def vowel(request):
    return request.param

@pytest.fixture(params=consonants)
def consonant(request):
    return request.param


def test(number, vowel, consonant):
    pass

Ваша интуиция была правильной. Путем параметризации каждого из нескольких приборов pytest заботится о создании всех возникающих перестановок.

Выход теста идентичен. Вот пример (я запустил py.test с опцией -vv):

test_bar.py:22: test[1-a-x] PASSED
test_bar.py:22: test[1-a-y] PASSED
test_bar.py:22: test[1-a-z] PASSED
test_bar.py:22: test[1-e-x] PASSED
test_bar.py:22: test[1-e-y] PASSED
test_bar.py:22: test[1-e-z] PASSED
test_bar.py:22: test[1-i-x] PASSED

Ответ 2

Вы можете применить несколько аргументов parametrize, и в этом случае они будут генерировать произведение всех параметров:

import pytest

numbers = [1,2,3,4,5]
vowels = ['a','e','i','o','u']
consonants = ['x','y','z']


@pytest.mark.parametrize('number', numbers)
@pytest.mark.parametrize('vowel', vowels)
@pytest.mark.parametrize('consonant', consonants)
def test(number, vowel, consonant):
    pass

Ответ 3

Я думаю, что помимо элегантного решения, вы также должны учитывать как количество времени, которое займет каждый параметр, так и объем кода, который вам придется поддерживать.

Возможные решения

1. Использование parametrize один раз с помощью itertools (предоставлено Frank T)
2. Использование 3 светильников (предоставлено Frank T)
3. Использование parametrize 3 раза (предоставлено Bruno Oliveira)
4. Использование 1 прибора и инструментов itertools (указано в вопросе)

Решение 1

@pytest.mark.parametrize('number, vowel, consonant',
                         itertools.product(numbers, vowels, consonants))
def test(number, vowel, consonant):
    pass

Решение 2

@pytest.fixture(params=numbers)
def number(request): return request.param

@pytest.fixture(params=vowels)
def vowel(request): return request.param

@pytest.fixture(params=consonants)
def consonant(request): return request.param


def test(number, vowel, consonant):
    pass

Решение 3

@pytest.mark.parametrize('number', numbers)
@pytest.mark.parametrize('vowel', vowels)
@pytest.mark.parametrize('consonant', consonants)
def test(number, vowel, consonant):
    pass

Решение 4

@pytest.fixture(params=cartesian)
def someparams(request):
  return request.param

def test_something(someparams):
  pass

Когда дело доходит до элегантности, я считаю, что Решение 3 - лучший вариант, потому что в нем меньше кода, и он не требует импорта itertools. После этого решение 1 является лучшим выбором, потому что вам не нужно писать фикстуры как решение 4 и решение 2. Решение 4, вероятно, лучше, чем Решение 2, потому что для его обслуживания требуется меньше кода.

Когда дело доходит до производительности, я запускаю каждое решение, используя numbers = list(range(100)), и получаю следующие результаты:

|  Solution  |  Time    | 
| Solution 1 |  3.91s   |
| Solution 2 |  3.59s   |
| Solution 3 |  3.54s   |
| Solution 4 |  3.09s   |