Pytest - 如何将参数传递给setup_class?

时间:2015-02-02 22:55:37

标签: python pytest

我有一些代码,如下所示。 我运行时遇到too few args错误。 我没有明确地调用setup_class,因此不确定如何将任何参数传递给它。 我尝试使用@classmethod修饰方法,但仍然看到相同的错误。

我看到的错误是这个 - E TypeError: setup_class() takes exactly 2 arguments (1 given)

需要注意的一点 - 如果我没有将任何参数传递给类,并且只传递cls,那么我没有看到错误。

非常感谢任何帮助。

我在发布之前检查了这些问题question #1question #2。我不明白这些问题的解决方案,或者它们如何运作。

class A_Helper:
    def __init__(self, fixture):
        print "In class A_Helper"

    def some_method_in_a_helper(self):
        print "foo"

class Test_class:
    def setup_class(cls, fixture):
        print "!!! In setup class !!!"
        cls.a_helper = A_Helper(fixture)

    def test_some_method(self):
        self.a_helper.some_method_in_a_helper()
        assert 0 == 0

2 个答案:

答案 0 :(得分:8)

你得到这个错误是因为你试图混合py.test支持的两种独立测试样式:经典的单元测试和pytest的固定装置。

我建议的不是混合它们,而是简单地定义一个类范围的夹具,如下所示:

import pytest

class A_Helper:
    def __init__(self, fixture):
        print "In class A_Helper"

    def some_method_in_a_helper(self):
        print "foo"

@pytest.fixture(scope='class')
def a_helper(fixture):
    return A_Helper(fixture)

class Test_class:
    def test_some_method(self, a_helper):
        a_helper.some_method_in_a_helper()
        assert 0 == 0

答案 1 :(得分:4)

由于你在pytest中使用它,它只会用一个参数和一个参数调用setup_class,看起来你不能改变它而不改变pytest calls this的方式。

您应该按照documentation并按照指定定义setup_class函数,然后使用您在该函数中需要的自定义参数在该方法中设置您的类,这看起来像< / p>

class Test_class:
    @classmethod
    def setup_class(cls):
        print "!!! In setup class !!!"
        arg = '' # your parameter here
        cls.a_helper = A_Helper(arg)

    def test_some_method(self):
        self.a_helper.some_method_in_a_helper()
        assert 0 == 0