pythonpython-unittestnose2

how to specify test specific setup and teardown in python unittest


I want to create unittest test with two different set up and tearDown methon in same class with two different test.

each test will use its specific setUp and tearDown method in python unittest framework.

could anyone help me.

    class processtestCase(unittest.TestCase):

         print "start the process test cases " 

        def setUp1(self): 
             unittest.TestCase.setUp(self)

        def test_test1(self): 
              "test Functinality" 

        def tearDown1(self): 
             unittest.TestCase.tearDown(self) 

        def setUp2(self): 
            unittest.TestCase.setUp2(self) 

        def test_test2(self): 
            "test Functinality" 

        def tearDown2(self): 
            unittest.TestCase.tearDown2(self) '

if __name__ == '__main__':
        unittest.main()

Solution

  • In the question, you mention that you have two tests, each with its own setup and teardown. There are at least two ways to go:

    You can either embed the setUp and tearDown code into each of the tests:

    class FooTest(unittest.TestCase):
        def test_0(self):
            ... # 1st setUp() code
            try:
                ... # 1st test code
            except:
                ... # 1st tearDown() code
                raise
    
        def test_1(self):
            ... # 2nd setUp() code
            try:
                ... # 2nd test code
            except:
                ... # 2nd tearDown() code
                raise
    

    Alternatively, you can split the class into two classes:

    class FooTest0(unittest.TestCase):
        @classmethod
        def setUp(cls):
            ...
    
        @classmethod
        def tearDown(cls):
            ...
    
        def test(self):
           ...
    

    The first option has fewer classes, is shorter, and more straightforsard. The second option more cleanly decouples setting up the fixture, and cleaning it up, then the test code itself. It also future proofs adding more tests.

    You should judge the tradeoffs based on your specific case, and your personal preferences.