本文为简单介绍，使用python自带模块unittest来进行单元测试
首先我们有一个需要测试的类，employee.py 定义了涨薪的方法。我们需要测试这个类的功能是否正确。
1 class Employee(): 2 3 def __init__(self,firstName,lastName,money): 4 self.firstName = firstName 5 self.lastName = lastName 6 self.money = money 7 8 def give_raise(self,add_money=5000): 9 self.money += add_money 10 return self.money
下面开始写单元测试模块 testEmployee.py
1 import unittest 2 from employee import Employee #导入需要测试的类 3 from HTMLTestRunner import HTMLTestRunner #导入生成HTML格式测试报告的模块。 4 5 class TestEmployee(unittest.TestCase): #需继承unittest.TestCase的类 6 7 def setUp(self): #每个测试用例执行前，需要做的操作 8 self.em = Employee('jade','chang',10000) #实例化一个雇员对象 9 10 def test_give_default_raise(self): #测试默认涨薪幅度，是否正确，用例名称需以test开头，否则不会运行。 11 self.assertEqual(self.em.give_raise(),15000) 12 13 def test_give_custom_raise(self): #测试自定义涨薪幅度 14 self.assertEqual(self.em.give_raise(2000),12000) 15 16 def report(): 17 filepath = r'C:\Users\Administrator\Desktop\RaiseMoney.html' 18 fp=open(filepath,'wb')
19 suite = unittest.TestSuite() #创建测试套件，将测试用例添加至套件中 20 suite.addTest(TestEmployee('test_give_default_raise')) 21 suite.addTest(TestEmployee('test_give_custom_raise'))
#生成测试报告 22 runner=HTMLTestRunner(stream=fp,title='雇员涨薪', description='雇员信息执行结果') 23 runner.run(suite) 24 unittest.main() 25 fp.close() 26 if __name__ == '__main__': 27 report()
运行成功后，生成的测试报告 RaiseMoney.html 如下：
点击Detail，可展开每条测试用例的执行情况。
unittest常见的断言方法：
| 断言方法 | 描述 |
| assertEqual(a,b) | a == b |
| assertNotEqual(a,b) | a != b |
| assertTrue(x) | x 为True |
| assertFalse(x) | x 为False |
| assertIn(item, list) | item in list |
| assertNotIn(item, list) | item not in list |