在pytest中使用参数化测试与参数的笛卡尔积

Question

我在想，有没有更优雅的方法来使用笛卡尔积进行参数化？这是我目前想到的：

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

cartesian = [elem for elem in itertools.product(*[numbers,vowels,consonants])]

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

def test_something(someparams):
  pass

至少我想把数字、元音字母、辅音字母和笛卡尔积都封装在这个测试函数里。

Answer 1

我觉得除了优雅的解决方案外，你还应该考虑每种选择所需的时间和你需要维护的代码量。

可能的解决方案

1. 使用一次 parametrize 和 itertools（由 Frank T 提供）
2. 使用 3 个 fixtures（由 Frank T 提供）
3. 使用 3 次 parametrize（由 Bruno Oliveira 提供）
4. 使用 1 个 fixture 和 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那样写 fixtures。解决方案 4可能比解决方案 2更好，因为它需要维护的代码更少。

在性能方面，我使用 numbers = list(range(100)) 运行了每个解决方案，得到了以下结果：

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

Answer 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

Answer 3

我想到两种方法来实现这个功能。一种是使用参数化的固定装置，另一种是对测试函数进行参数化。你可以选择你觉得更优雅的那种。

下面是参数化的测试函数：

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会自动处理所有可能的组合。

测试输出是一样的。这里有一个示例（我用 -vv 选项运行了 py.test）：

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