Python单元测试unittest与HTMLTestRunner报告生成

本文为简单介绍,使用python自带模块unittest来进行单元测试

首先我们有一个需要测试的类,employee.py  定义了涨薪的方法。我们需要测试这个类的功能是否正确。

class Employee():
    
    def __init__(self,firstName,lastName,money):
        self.firstName = firstName
        self.lastName = lastName
        self.money = money
    
    def give_raise(self,add_money=5000):
        self.money += add_money
        return self.money

下面开始写单元测试模块  testEmployee.py

import unittest
from employee import Employee  #导入需要测试的类
from HTMLTestRunner import HTMLTestRunner  #导入生成HTML格式测试报告的模块。

class TestEmployee(unittest.TestCase):  #需继承unittest.TestCase的类
    
    def setUp(self):  #每个测试用例执行前,需要做的操作
        self.em = Employee(‘jade‘,‘chang‘,10000)  #实例化一个雇员对象
        
    def test_give_default_raise(self):  #测试默认涨薪幅度,是否正确,用例名称需以test开头,否则不会运行。
        self.assertEqual(self.em.give_raise(),15000)
    
    def test_give_custom_raise(self):  #测试自定义涨薪幅度
        self.assertEqual(self.em.give_raise(2000),12000)

def report():
    filepath = r‘C:\Users\Administrator\Desktop\RaiseMoney.html‘ 
    fp=open(filepath,‘wb‘)19     suite = unittest.TestSuite()  #创建测试套件,将测试用例添加至套件中
    suite.addTest(TestEmployee(‘test_give_default_raise‘))
    suite.addTest(TestEmployee(‘test_give_custom_raise‘))      #生成测试报告
    runner=HTMLTestRunner(stream=fp,title=‘雇员涨薪‘, description=‘雇员信息执行结果‘)
    runner.run(suite)
    unittest.main()
    fp.close()
if __name__ == ‘__main__‘:
    report()

运行成功后,生成的测试报告 RaiseMoney.html 如下:

点击Detail,可展开每条测试用例的执行情况。

Python单元测试unittest与HTMLTestRunner报告生成

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

相关推荐