Python單元測試unittest
編輯推薦: |
本文來自cnblogs,本文介紹了Python中自帶的單元測試框架是unittest模組的相關內容。 |
Python中有一個自帶的單元測試框架是unittest模組,用它來做單元測試,它裡面封裝好了一些校驗返回的結果方法和一些用例執行前的初始化操作。
在說unittest之前,先說幾個概念:
TestCase 也就是測試用例
TestSuite 多個測試用例集合在一起,就是TestSuite
TestLoader是用來載入TestCase到TestSuite中的
TestRunner是來執行測試用例的,測試的結果會儲存到TestResult例項中,包括運行了多少測試用例,成功了多少,失敗了多少等資訊
下面寫一個簡單的單元測試用例
import unittest class MyTest(unittest.TestCase): # 繼承unittest.TestCase def tearDown(self): # 每個測試用例執行之後做操作 print('111') def setUp(self): # 每個測試用例執行之前做操作 print('22222') @classmethod def tearDownClass(self): # 必須使用 @ classmethod裝飾器, 所有test執行完後執行一次 print('4444444') @classmethod def setUpClass(self): # 必須使用@classmethod 裝飾器,所有test執行前執行一次 print('33333') def test_a_run(self): self.assertEqual(1, 1) # 測試用例 def test_b_run(self): self.assertEqual(2, 2) # 測試用例 if __name__ == '__main__': unittest.main()#執行所有的測試用例
下面是一些常用的斷言,也就是校驗結果
assertEqual(a, b) a == b assertNotEqual(a, b) a != b assertTrue(x) bool(x) is True assertFalse(x) bool(x) is False assertIsNone(x) x is None assertIsNotNone(x) x is not None assertIn(a, b) a in b assertNotIn(a, b) a not in b
那如何生成一個測試報告呢,需要加入另外一個模組了,HTMLTestRunner,這個模組需要自己安裝,使用執行測試用例就會生成一個html的測試報告,裡面會有每個測試用例的執行結果,程式碼如下:
import HTMLTestRunner import unittest class MyTest(unittest.TestCase):#繼承unittest.TestCase def tearDown(self): #每個測試用例執行之後做操作 print('111') def setUp(self): #每個測試用例執行之前做操作 print(22222) def test_run(self): # self.assertEqual(1,1) self.assertIs(1,1) #測試用例 def test_run2(self): # self.assertEqual(1,1) self.assertIs(1,1) #測試用例 def test_run3(self): # self.assertEqual(1,1) self.assertIs(1,1) #測試用例 def test_run1(self): # self.assertEqual(1,1) self.assertIs(1,1) #測試用例 if __name__ == '__main__': test_suite = unittest.TestSuite()#建立一個測試集合 test_suite.addTest(MyTest('test_run1'))#測試套件中新增測試用例 #test_suite.addTest(unittest.makeSuite(MyTest))#使用makeSuite方法新增所有的測試方法 fp = open('res.html','wb')#開啟一個儲存結果的html檔案 runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='api測試報告',description='測試情況') #生成執行用例的物件 runner.run(test_suite) #執行測試套件
如果我們有很多個模組,每個模組下面都寫了很多python檔案,每個python檔案裡面都有測試用例,那怎麼把這個目錄下的用例都執行了呢,就要先找到這個目錄下的所有python檔案,然後找到裡面的測試用例,逐個執行,程式碼如下:
import unittest,HTMLTestRunner suite = unittest.TestSuite()#建立測試套件 all_cases = unittest.defaultTestLoader.discover('.','test_*.py') #找到某個目錄下所有的以test開頭的Python檔案裡面的測試用例 for case in all_cases: suite.addTests(case)#把所有的測試用例新增進來 fp = open('res.html','wb') runner = HTMLTestRunner.HTMLTestRunner(stream=fp, title='all_tests',description='所有測試情況') runner.run(suite) #執行測試
我們在後續進行持續整合的時候,要讓程式碼自動執行,就會用到Jenkins了,但是上面產生的測試報告都是html格式的,Jenkins不認識,就在Jenkins裡面顯示不出來。那咱們就要產生一些Jenkins認識的測試報告,Jenkins認識xml格式的報告,那咱們就產生xml格式的唄,就需要用一個新的模組,xmlrunner,安裝直接 pip install xmlrunner即可,程式碼如下:
import unittest import xmlrunner #匯入這個模組 class My(unittest.TestCase): def test1(self,a,b,c): self.assertEqual(a+b,c) if __name__=='__main__': test_suite = unittest.TestSuite() test_suite.addTest(unittest.makeSuite(My)) runner = xmlrunner.XMLTestRunner(output='report')#指定報告放的目錄 runner.run(test_suite)
然後咱們執行,可以看到在report目錄下已經產生了xml格式的報告了,而且還自動把日期加上了