__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

Установите unittest2 и используйте этот пакет unittest.

import unittest2 

, а затем использовать setupModule/tearDownModule или setupClass/класс Teardown для специальной логики инициализации

Подробнее: http://www.voidspace.org.uk/python/articles/unittest2.shtml

Кроме того, скорее всего, ваш создаете тест интеграции больше, чем UnitTest. Выберите хорошее имя для тестов, чтобы отличить их или положить в другой контейнерный модуль.