Python unittest -- 在我的 setUpClass 方法中使用父类方法？

Question

I am writing unit tests and have the following structure:

class TestSuite(unittest.TestCase, HelperClass):
  @classmethod
  def setUpClass(cls):
    # I want to use methods from HelperClass here, but get errors
    # I have tried cls.method_name() and self.method_name()

  def setUp(self):
    self.method_name() # methods from HelperClass work here
...

As noted in my comments, I'd like to use some validation methods from HelperClass in my setUpClass method, but if I try to call with cls.method_name(), I get TypeError: method_name() missing 1 required positional argument: 'self' , but if I use self.method_name(), I get NameError: name 'self' is not defined .

This is probably something pretty basic, but I'm just not sure the correct search term I'm supposed to use to find the answer. The unittest documentation on setUpClass doesn't get into it either, unfortunately.

Answer 1

The problem is that in setUpClass , you simply don't have an instance of TestSuite on which you can call instance methods defined by HelperClass . The test runner does something along the lines of

TestSuite.setUpClass()

t = TestSuite(...)

t.setUp()
t.test_foo()
t.tearDown()

t.setUp()
t.test_bar()
t.tearDown()

setUpClass is called once , without an instance of TestSuite , before you create the single instance that will be used to call each test defined in the class. The setUp and tearDown methods are called before and after, respectively, each test.

What you might want is to override TestSuite.__init__ , which is called once per instance, rather than before each test method is called.

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

Answer 2

I don't know if it was the right solution, but I found a way to do this that was ultimately pretty obvious.

class TestSuite(unittest.TestCase, HelperClass):
  @classmethod
  def setUpClass(cls):
    hc = HelperClass()
    hc.method_name()

  def setUp(self):
    self.method_name() # methods from HelperClass work here
...

In retrospect, I should have seen this right away, haha. Figured I'd share for someone also searching around on this someday.