Python Unittest

Posted on Edited on In Word count in article: 559 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
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 setUp()--->test_XXX()--->tearDown()--->setUp()--->test_XXX()--->tearDown().....