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()