使用unittest
这个包,官方文档:https://docs.python.org/zh-cn/3/library/unittest.html
举个栗子:
写函数
在add_code.py
这个文件中先写一个函数:
def append(a, b):
return a + b
写单元测试
要测试上面的append()
,使用下面的代码:
import unittest
class AddCodeTest(unittest.TestCase):
def test_add_code(self):
import add_code
self.assertEqual(add_code.append(1, 2), 3) # 使用 self.assert.... 来做判别
既可以直接运行AddCodeTest
类,也可以直接运行test_add_code
函数,日志显示为:
Ran 1 test in 0.002s
Launching unittests with arguments python -m unittest test_add_code.AddCodeTest in my_test
OK
注意:测试类所在的包叫tests