使用pytest

时间:2016-08-02 19:31:47

标签: python python-3.x pytest

我需要为测试类中的每个测试运行一系列对象。我想参数化TestClass中的每个测试函数。最终目标是有类似的东西:

@pytest.mark.parametrize('test_input', [1, 2, 3, 4])
class TestClass:
    def test_something1(self, test_input):
        # test code here, runs each time for the parametrize

但根据我的理解,您无法传递输入参数,或至少在课堂上调用@pytest.mark.parametrize,这些标记适用于defs而不是class s

我现在拥有的:

class TestClass:
    def test_something1(self):
        for i in stuff:
            # test code here

    def test_something2(self):
        for i in stuff:
            # test code here
    ...

有没有办法传递一个类本身或TestClass中的每个函数的参数化?@pytest.mark.parametrize

中可能只有一个@pytest.fixture...(autouse=True).

我希望将我的测试组织成一个类,因为它反映了我正在测试的文件。因为我在至少十几个不同的测试中遍历这些对象,所以调用类的循环比在每个def中更容易。

2 个答案:

答案 0 :(得分:1)

我已经解决了。我太复杂了;而不是使用标记我可以使用传递参数的fixture函数。

在我找到答案之前(没有参数化):

class TestClass:
    def test_something(self):
        for i in example_params:
            print(i)

回答,使用pytest fixture。会做同样的事情,但只需要输入,而不是for循环:

import pytest
example_params = [1, 2, 3]

@pytest.fixture(params=example_params)
def param_loop(request):
    return request.param

class TestClass:
    def test_something(self, param_loop):
        print(param_loop)

因此,要参数化所有def s:

  1. 使用@pytest.fixture(params=[])
  2. 上的def my_function(request)装饰器
  3. 内部my_functionreturn request.param
  4. my_function添加到要参数化的任何函数的输入

答案 1 :(得分:0)

您可以将parametrize应用于课程。来自the docs

  

@pytest.mark.parametrize允许在测试功能或上定义多组参数和固定装置。

相同的数据将发送到该类中的所有测试方法。

@pytest.mark.parametrize('test_input', [1, 2, 3, 4])
class TestClass:
    def test_something1(self, test_input):
        pass


    def test_something2(self, test_input):
        pass

如果使用pytest -v使用以下命令运行测试,则会得到以下输出:

=========================================================================================== test session starts ============================================================================================
platform darwin -- Python 3.7.3, pytest-4.4.2, py-1.8.0, pluggy-0.11.0 -- /Users/user/.local/share/virtualenvs/stack-overlfow-pycharm-L-07rBZ9/bin/python3.7
cachedir: .pytest_cache
rootdir: /Users/user/Documents/spikes/stack-overlfow-pycharm
collected 8 items

test_class.py::TestClass::test_something1[1] PASSED                                                                                                                                                  [ 12%]
test_class.py::TestClass::test_something1[2] PASSED                                                                                                                                                  [ 25%]
test_class.py::TestClass::test_something1[3] PASSED                                                                                                                                                  [ 37%]
test_class.py::TestClass::test_something1[4] PASSED                                                                                                                                                  [ 50%]
test_class.py::TestClass::test_something2[1] PASSED                                                                                                                                                  [ 62%]
test_class.py::TestClass::test_something2[2] PASSED                                                                                                                                                  [ 75%]
test_class.py::TestClass::test_something2[3] PASSED                                                                                                                                                  [ 87%]
test_class.py::TestClass::test_something2[4] PASSED                                                                                                                                                  [100%]

========================================================================================= 8 passed in 0.03 seconds =========================================================================================

这正是您想要的。