在阅读源码的时候遇到了unittest.main()函数,网上查到用法,做些记录

  需要导入其模块

import unittest # 导入unittest模块

  unittest模块用于单元测试.其中测试类需要传入参数unittest.TestCase

class UserCase(unittest.TestCase): # 需要传入unittest.TestCase
    pass

  并且类中的函数需要以test开头,方可执行.

class UserCase(unittest.TestCase):

    def testAddUser(self): #以test开头
        print("add a user")

  最后是调用:unittest.main()

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

完整测试代码

import unittest

class UserCase(unittest.TestCase):

    def testAddUser(self):
        print("add a user")

    def testDelUser(self):
        print("delete a user")

class haha(unittest.TestCase):
    def test_haha(self):
        print('我是一个测试')
    def a(self): # 这一个函数是不会执行的
        print('我不是一个测试')

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

参考

• unittest之入口main()