Python unittest測試框架詳解

概述

unittest測試框架包括如下：

1.測試腳手架(test fixture)

測試準備前要做的工作和測試執行完後要做的工作.包括setUp()和tearDown().

2.測試案例(test case)

最小的測試單元.

3.測試套件(test suite)

測試案例的集合.

4.測試運行器(test runner)

測試執行的組件.

命令行接口

可以用命令行運行測試模塊,測試類以及測試方法.

複製代碼代碼如下:

python -m unittest test_module1 test_module2
python -m unittest test_module.TestClass
python -m unittest test_module.TestClass.test_method

可加-v打印詳細信息

複製代碼代碼如下:

python -m unittest -v test_module

測試案例自動搜索

unittest支持簡單的test discovery. 命令行傳入discovery後,框架會自動在當前目錄搜索要測試的案例並執行.搜索目錄必須是包或者模塊.基本使用如下:

複製代碼代碼如下:

cd project_directory 
python -m unittest discover

子選項如下: 
-v, –verbose 
輸出信息的詳細級別

-s, –start-directory directory 
開始搜索目錄 (默認爲當前目錄)

-p, –pattern pattern 
匹配的文件名 (默認爲test*.py)

-t, –top-level-directory directory 
搜索的頂層目錄 (默認爲start directory)

創建測試代碼

1.方式一

創建子類繼承unittest.TestCase,然後重寫以下方法

複製代碼代碼如下:

class WidgetTestCase(unittest.TestCase):
    def setUp(self):
        pass
    def runTest(self):
        pass
    def tearDown(self):
        pass

運行

2.方式二

編寫以test開頭的方法

複製代碼代碼如下:

class WidgetTestCase(unittest.TestCase):
    def setUp(self):
        pass

    def test_xx1(self)
    def test_xx2(self)
    ...
    def test_xxN(self)

    def tearDown(self):
        pass

構建測試套件

方式一

複製代碼代碼如下:

widgetTestSuite = unittest.TestSuite()
widgetTestSuite.addTest(WidgetTestCase('test_default_size'))
widgetTestSuite.addTest(WidgetTestCase('test_resize'))

方式二(推薦)

複製代碼代碼如下:

def suite():
    suite = unittest.TestSuite()
    suite.addTest(WidgetTestCase('test_default_size'))
    suite.addTest(WidgetTestCase('test_resize'))
    return suite

方式三(推薦)

複製代碼代碼如下:

def suite():
    tests = ['test_default_size', 'test_resize']
    return unittest.TestSuite(map(WidgetTestCase, tests))

其中還有幾種方式不常用。

待測試的類(Widget.py)

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

# Widget.py # Python 2.7.6    class Widget:   def __init__(self, size = (40,40)):     self.size = size   def getSize(self):     return self.size   def reSize(self,width,height):     if width <0 or height < 0:       raise ValueError, 'illegal size'     else:       self.size = (width,height)       return self.size   def dispose(self):     pass

測試類(Auto.py)

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44

# coding=utf8 # Auto.dy # Python 2.7.6    from Widget import Widget #導入測試類模塊Widget import unittest      #導入unittest模塊        class WidgetTestCase(unittest.TestCase): #讓所有執行測試的類都繼承於TestCase類，可以將TestCase看成是對特定類進行測試的方法的集合      #在setUp()方法中進行測試前的初始化工作。   def setUp(self):       self.widget = Widget()      #並在tearDown()方法中執行測試後的清除工作，setUp()和tearDown()都是TestCase類中定義的方法。   def tearDown(self):     self.widget = None      #測試Widget類中getSize方法   def testgetSize(self):     print "Test GetSize"     #對Widget類中getSize()方法的返回值和預期值進行比較，確保兩者是相等的，     #assertEqual()也是TestCase類中定義的方法。     self.assertEqual(self.widget.getSize(), (40, 40))          #測試Widget類中reSize方法   def testreSize(self):     print "Test Resize"     #對Widget類中reSize()方法的返回值和預期值進行比較，確保兩者是相等的。     #assertEqual()也是TestCase類中定義的方法。     self.assertEqual(self.widget.reSize(50,100),(50,100))         #提供名爲suite()的全局方法，PyUnit在執行測試的過程調用suit()方法來確定有多少個測試用例需要被執行， #可以將TestSuite看成是包含所有測試用例的一個容器。 def suite():   suite = unittest.TestSuite()   suite.addTest(WidgetTestCase("testgetSize"))#往此添加需要測試的方法testgetSize()   suite.addTest(WidgetTestCase("testreSize")) #往此添加需要測試的方法testreSize()   return suite     if __name__ == "__main__":   unittest.main(defaultTest = 'suite') #在主函數中調用全局方法.

測試結果:

?

1 2 3 4 5 6 7 8

D:\Python>python27 Auto.py Test GetSize .Test Resize . ------------------------------ Ran 2 tests in 0.004s    OK

總結:
1。第一步:先寫好測試類
2。第二步:導入unittest模塊及測試的類，運用setup()方法做測試前的準備工作，如建立數據庫連接，運用teardown()方法做測試後的清除工作，如取消數據庫的鏈接，再對類中的方法逐一做測試。
3。第三步: 寫suite()的全局方法，將要測試的方法，一一加入。
測試結果，有幾個測試用例就有幾個. 最後顯示OK，表示通過。