__init__为unittest.TestCase

Question

我想unittest.TestCase在初始化时给类添加一些东西,但我无法弄清楚如何去做.

现在我这样做:

#filename test.py

class TestingClass(unittest.TestCase):

    def __init__(self):
        self.gen_stubs()

    def gen_stubs(self):
        # Create a couple of tempfiles/dirs etc etc.
        self.tempdir = tempfile.mkdtemp()
        # more stuff here

我希望在整个测试集中只生成一次所有存根.我无法使用,setUpClass()因为我正在使用Python 2.4(我无法在python 2.7上工作).

我在这做错了什么？

我收到此错误:

 `TypeError: __init__() takes 1 argument (2 given)` 

... __init__当我使用命令运行它时,我将所有存根代码移动到其他错误python -m unittest -v test.

Answer 1

试试这个:

class TestingClass(unittest.TestCase):

    def __init__(self, *args, **kwargs):
        super(TestingClass, self).__init__(*args, **kwargs)
        self.gen_stubs()

您正在覆盖TestCase的__init__,所以你可能想要让基类处理这些参数为您服务.

Answer 2

Just wanted to add some clarifications about overriding the init function of

unittest.TestCase

The function will be called before each method in your test class. Please note that if you want to add some expensive computations that should be performed once before running all test methods please use the SetUpClass classmethod

@classmethod
def setUpClass(cls):
    cls.attribute1 = some_expensive_computation()

This function will be called once before all test methods of the class. See setUp for a method that is called before each test method.