Python中有一個自帶的單元測試框架是unittest模塊,用它來做單元測試,它里面封裝好了一些校驗返回的結果方法和一些用例執行前的初始化操作。
unittest是python的標準測試庫,相比于其他測試框架是python目前使用最廣的測試框架。
unittest有四個比較重要的概念是test fixture, test case, test suite, test runner, 。
在說unittest之前,先說幾個概念:
TestCase 也就是測試用例
TestSuite 多個測試用例集合在一起,就是TestSuite
TestLoader是用來加載TestCase到TestSuite中的
TestRunner是來執行測試用例的,測試的結果會保存到TestResult實例中,包括運行了多少測試用例,成功了多少,失敗了多少等信息
通過dir(unittest),我們可以看到unittest全部的屬性和方法,他們的關系如下圖所示。

unittest主要類關系
正常調用unittest的流程是,TestLoader 自動將測試用例TestCase中加載到TestSuite里,TextTestRunner調用TestSuite的run方法,順序執行里面的TestCase中以test開頭的方法,并得到測試結果TestResult。在執行TestCase過程中,先進行SetUp()環境準備,執行測試代碼,最后tearDown()進行測試的還原。
其中TestLoader在加載過程中,進行添加的TestCase是沒有順序的。一個TestCase里如果存在多個驗證方法的話,會按照方法中test后方首字母的ascii碼從小到大排序后執行。
可以通過手動調用TestSuite的addTest、addTests方法來動態添加TestCase,這樣既可以確定添加用例的執行順序,也可避免TestCase中的驗證方法一定要用test開頭。
下面寫一個簡單的單元測試用例
import unittestclass 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
新聞熱點
疑難解答