Python Unittest

Posted on Edited on Word count in article: 562 Reading time ≈ 1 mins.
first you should import unittest, and write a class inherit unittest.TestCase. Then write some function named test_XXX, which uses "test_" as its prefix. Then in main, call the class's main() function.

this is an example:

1
2
3
4
5
6
7
8
9
10
import unittest

class TestFunctions(unittest.TestCase):    
    def test_range(self):
       for i in range(10):
          self.assertEqual(i,i)

if __name__ == "__main__":
    TestFunctions.main()

In this example, TestFunctions.main() will call all functions like test_XXX between setUp() and tearDown(). The sequence is 
1
setUp()--->test_XXX()--->tearDown()--->setUp()--->test_XXX()--->tearDown().....