python unittest生成html报告_Python单元测试unittest与HTMLTestRunner报告生成

本文为简单介绍，使用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:UsersAdministratorDesktopRaiseMoney.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

备注：

HTMLTestRunner.py 下载链接：

链接：https://pan.baidu.com/s/1FIt_6wOVCF7ZxWSUD95OSA

提取码：44hu