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

    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，可展開每條測試用例的執行情況。

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