pytest动态参数化测试

时间:2018-06-12 13:01:05

标签: python pytest

我创建了一个'动态'参数列表,我将其传递给参数化。

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),我想继续使用词典将数据传递给测试。

2 个答案:

答案 0 :(得分:2)

  1. 评论中提到的Paulo Scardine,请勿重新发明轮子,请使用itertools.combinations
  2. 您正在寻找ids argument in parametrize hook。它是每个参数的名称列表,或者是接受参数并返回字符串的函数 - repr是内置类型的简单解决方案。
  3. 组合示例:

    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
    

答案 1 :(得分: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 ================================

[更新]

  

我认为这个答案是我将使用的解决方案。每一天都是上学日! (出于好奇,有一种方式我可以成为' param'变成像第一个,第二个'所以让test_foo(param)成为test_foo(第一个,第二个)。我&# 39;我不确定实际上有什么帮助...但我很好奇 - 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进行测试,因此大多数时候输出只是每个测试的一个点。