pytest 动态参数化测试

Question

我创建了一个“动态”参数列表，我将其传递给参数化。

OPTIONS = ['a', 'b', 'c']

def get_unique_pairs():
    unique_list = []
    for first in OPTIONS:
        for second OPTIONS:
            if first == second:
                continue
            unique_list.append({'first':first, 'second':second))
    return unique_list

def some_func()
    unique_pairs = get_unique_pairs()
    result = []
    for pair in unique_pair:
        if test(pair):
           continue
        else:
           result.append(pair)
    return pair

@pytest.mark.parametrize('param', some_fnc())
def test_fnc(param):
    first = param['first']
    second = param['second']

我希望传递给test_fnc 的输入是[('a','b'),('a','c')...('c','b')]，其中第一个和第二个元素永远不会相同。我正在使用一些额外的逻辑来进一步删除特定的对。

当我运行测试时，我得到了输出：

::test_fnc[param0] PASSED
::test_fnc[param1] PASSED
::test_fnc[param2] PASSED

我有两个问题：

1. 我不完全确定如何描述我正在做什么以查找更多文档/帮助。
2. 我想要更多描述性的输出（即不是param0），并且我想继续使用字典将数据传递给测试。

Answer 1

我会这样写：

import pytest
import itertools


OPTIONS = ['a', 'b', 'c']

@pytest.mark.parametrize(
    'param',
    itertools.permutations(OPTIONS, 2),
    ids=lambda pair: "first={}, second={}".format(*pair)
)
def test_fn(param):
    first, second = param
    assert first != second

结果：

> pytest --verbose 
================================ test session starts =================================
platform linux2 -- Python 2.7.12, pytest-3.6.1, py-1.5.3, pluggy-0.6.0 -- /usr/bin/python
cachedir: .pytest_cache
rootdir: /home/paulos/work/lixo/tests, inifile:
collected 6 items

test_foo.py::test_fn[first=a, second=b] PASSED                                     [ 16%]
test_foo.py::test_fn[first=a, second=c] PASSED                                     [ 33%]
test_foo.py::test_fn[first=b, second=a] PASSED                                     [ 50%]
test_foo.py::test_fn[first=b, second=c] PASSED                                     [ 66%]
test_foo.py::test_fn[first=c, second=a] PASSED                                     [ 83%]
test_foo.py::test_fn[first=c, second=b] PASSED                                     [100%]

================================ 6 passed in 0.03 seconds ================================

[更新]

我认为这个答案是我将使用的解决方案。每一天都是上学日！ （出于好奇，有没有一种方法可以将“参数”变成“第一，第二”之类的东西，所以让 test_foo（param）成为 test_foo（第一，第二）。我不确定这是否真的有帮助......但我很好奇——F. Elliot

如果您不介意test_fn[a-b] 而不是test_fn[a, b]：

@pytest.mark.parametrize(
    'first,second',
    itertools.permutations(OPTIONS, 2),
)
def test_fn(first, second):
    assert first != second

实际上，我们并没有真正使用--verbose 运行测试，所以大多数时候每个测试的输出只是一个点。

Answer 2

1. 正如评论中提到的Paulo Scardine，不要重新发明轮子，使用itertools.combinations。
2. 您正在寻找ids argument in parametrize hook。它可以是每个参数的名称列表，也可以是接受参数并返回字符串的函数 - repr 是内置类型的简单解决方案。

组合示例：

import itertools
import pytest


opts = ['a', 'b', 'c']


@pytest.mark.parametrize('param', itertools.combinations(opts, 2), ids=repr)
def test_fn(param):
    first = param[0]
    second = param[1]
    assert first != second