问题背景
在阅读源码的时候遇到了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()