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

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