python單元測試unittest實例詳解

更新時間：2015年05月11日 11:13:17 作者：像風一樣的自由

這篇文章主要介紹了python單元測試unittest用法,以實例形式詳細分析了Python中單元測試的概念、用法與相關使用技巧,具有一定參考借鑒價值,需要的朋友可以參考下

本文實例講述了python單元測試unittest用法。分享給大家供大家參考。具體分析如下：

單元測試作為任何語言的開發(fā)者都應該是必要的，因為時隔數月后再回來調試自己的復雜程序時，其實也是很崩潰的事情。雖然會很快熟悉內容，但是修改和調試將是一件痛苦的事情，如果你在修改了代碼后出現問題的話，而單元測試可以幫助我們很快準確的定位到問題的位置，出現問題的模塊和單元。所以這是一件很愉快的事情，因為我們知道其它修改或沒有修改的地方仍然是正常工作的，而我們目前的唯一問題就是搞定眼前這個有點問題的“家伙”。所以工作會在輕松中開始，并且很快將會結束，因為你已經知道很多信息了。

單元測試自然是對程序中最小的可測試模塊--函數來進行測試；因為單元測試的對象是函數，也就是說你得被測試對象一定要有輸出結果，哪怕就是異常輸出，也得有輸出，以便單元測試模塊能夠捕獲返回值，并且與預期值進行比較，從而得出測試通過與否。

單元測試的加載方式有2種：一種是通過unittest.main()來啟動單元測試的測試模塊；一種是添加到testsuite集合中再加載所有的被測試對象，而testsuit里存放的就是單元測試的用例，下面分別列出了2種方法的使用。

1.1 測試模塊中的函數：

被測模塊：

#!/usr/bin/env python 
#encoding: utf-8 
def sum( x, y): 
    return x+y 
def sub( x, y): 
    return x-y

單元測試模塊：

#!/usr/bin/env python 
#encoding: utf-8 
import unittest 
import myclass 
class mytest(unittest.TestCase): 
  ##初始化工作 
  def setUp(self): 
    pass 
  #退出清理工作 
  def tearDown(self): 
    pass 
  #具體的測試用例，一定要以test開頭 
  def testsum(self): 
    self.assertEqual(myclass.sum(1, 2), 2, 'test sum fail') 
  def testsub(self): 
    self.assertEqual(myclass.sub(2, 1), 1, 'test sub fail')   
if __name__ =='__main__': 
  unittest.main()

測試結果：【F表示一個fail， F前的點表示一個通過，有E的話表示程序自身異?！?/p>

.F
======================================================================
FAIL: testsum (__main__.mytest)
----------------------------------------------------------------------
Traceback (most recent call last):
 File "C:\Users\xiaowu\workspace\mypython\unitTest.py", line 19, in testsum
  self.assertEqual(myclass.sum(1, 2), 2, 'test sum fail')
AssertionError: test sum fail
----------------------------------------------------------------------
Ran 2 tests in 0.001s 
FAILED (failures=1)

1.2 測試模塊類中的函數：

被測模塊：

#!/usr/bin/env python
#encoding: utf-8
class myclass:
  def __init__(self):
    pass
  def sum(self, x, y):
    return x+y
  def sub(self, x, y):
    return x-y

單元測試模塊：

#!/usr/bin/env python 
#encoding: utf-8 
import unittest 
import myclass 
class mytest(unittest.TestCase): 
  ##初始化工作 
  def setUp(self): 
    self.tclass = myclass.myclass()
    ##實例化了被測試模塊中的類 
  #退出清理工作 
  def tearDown(self): 
    pass 
  #具體的測試用例，一定要以test開頭 
  def testsum(self): 
    self.assertEqual(self.tclass.sum(1, 2), 3) 
if __name__ =='__main__': 
  unittest.main()

運行結果：

.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK

這種方式執(zhí)行單個測試文件時使用-v參數可以獲得更多的測試結果信息。如：mytest.py -v

2 加載測試套件

好吧，在運用測試套件進行單元測試之前，我想還是稍微研究一下unittest模塊的內容有哪些，其大概的運行方式是什么樣的。而后在給出根據各種情況如何制定單元測試套件。

首先，自然是查看unittest模塊有哪些成員啦！

>>import unittest 
>>dir(unittest) 
['FunctionTestCase', 'TestCase', 'TestLoader', 'TestProgram', 'TestResult', 'Tes 
tSuite', 'TextTestRunner', '_CmpToKey', '_TextTestResult', '_WritelnDecorator', 
'__all__', '__author__', '__builtins__', '__doc__', '__email__', '__file__', '__ 
metaclass__', '__name__', '__package__', '__unittest', '__version__', '_makeLoad 
er', '_strclass', 'defaultTestLoader', 'findTestCases', 'getTestCaseNames', 'mai 
n', 'makeSuite', 'os', 'sys', 'time', 'traceback', 'types']

可以看到其自身的成員也不是很多，大概包括有：

['FunctionTestCase', 'TestCase', 'TestLoader', 'TestProgram', 'TestResult',
 'TestSuite','TextTestRunner', '_CmpToKey', '_TextTestResult', '_WritelnDecorator',
 'defaultTestLoader','findTestCases', 'getTestCaseNames', 'main', 'makeSuite']

好吧,我們大概看看具體都是干什么的

>>memblist = ['FunctionTestCase', 'TestCase', 'TestLoader', 'TestProgram', 'TestResult',\ 
'TestSuite','TextTestRunner', 'defaultTestLoader','findTestCases', 'getTestCaseNames', \ 
'main', 'makeSuite'] 
>>for memb in memblist: 
.. cur = getattr(unittest, memb) 
.. print help(cur)

'FunctionTestCase':函數測試用例，即給一個函數作為參數，返回一個testcase實例，可選參數有set-up，tear-down方法
'TestCase'：所有測試用例的基本類，給一個測試方法的名字，返回一個測試用例實例
'TestLoader'：測試用例加載器，其包括多個加載測試用例的方法。返回一個測試套件
loadTestsFromModule(self, module)--根據給定的模塊實例來獲取測試用例套件
loadTestsFromName(self, name, module=None)
--根據給定的字符串來獲取測試用例套件，字符串可以是模塊名，測試類名，測試類中的測試方法名，或者一個可調用的是實例對象
這個實例對象返回一個測試用例或一個測試套件
loadTestsFromNames(self, names, module=None) --和上面功能相同，只不過接受的是字符串列表
loadTestsFromTestCase(self, testCaseClass)--根據給定的測試類，獲取其中的所有測試方法，并返回一個測試套件
'TestProgram'：命令行進行單元測試的調用方法，作用是執(zhí)行一個測試用例。其實unittest.main()方法執(zhí)行的就是這個命令，
而這個類實例時默認加載當前執(zhí)行的作為測試對象，
原型為 __init__(self, module='__main__', defaultTest=None, argv=None, testRunner=xx, testLoader=xx)
其中module='__main__'就是默認加載自身
'TestResult'：測試用例的結果保存實例，通常有測試框架調用
'TestSuite'：組織測試用例的實例，支持測試用例的添加和刪除，最終將傳遞給testRunner進行測試執(zhí)行
'TextTestRunner'：進行測試用例執(zhí)行的實例，其中Text的意思是以文本形式顯示測試結果。顯示測試名稱，即完成的測試結果，其過同執(zhí)行單元測試腳本時添加-v參數
'defaultTestLoader':其實就是TestLoader
'findTestCases', 'getTestCaseNames'：這個2個就不用解釋了
'main': 其實就是TestProgram
'makeSuite'：通常是由單元測試框架調用的，用于生產testsuite對象的實例

至此，我們知道了。其實整個單元測試框架的邏輯出來了。分三步走：第一步testloader根據傳入的參數獲得相應的測試用例，即對應具體的測試方法，
然后makesuite在把所有的測試用例組裝成testsuite，最后把testsiute傳給testrunner進行執(zhí)行。
而我們通常執(zhí)行的unittest.main()，其實就是unittest.testprom方法，其執(zhí)行的功能就是上面分析的三步，在第一步中其傳入的參數是自身的模塊__main__；
在第二步中把自身模塊中的所有測試類中中的測試方法提取出來，并生成測試套件；最后再把測試套件傳遞給testrunner進行具體的測試。

最后給出一個完整的單元測試組織代碼，把該代碼放到單元測試用例文件的同一個目錄后執(zhí)行該腳本，即可執(zhí)行所有的測試用例文件。

【測試用例文件必須為test開頭，如：testxxx.py, 當然這個文件本身是一個單元測試的文件】

#!/usr/bin/env python
#encoding: utf-8
#該代碼源自深入python
import unittest
import myclass
import re
import os
import sys
def testAllinCurrent():
  path = os.path.abspath(os.path.dirname(sys.argv[0]))
  files = os.listdir(path)
  test = re.compile("test\.py{1}quot;, re.IGNORECASE)
  files = filter(test.search, files)
  filenameToModuleName = lambda f: os.path.splitext(f)[0]
  moduleNames = map(filenameToModuleName, files)
  modules = map(__import__, moduleNames)
  load = unittest.defaultTestLoader.loadTestsFromModule
  return unittest.TestSuite(map(load, modules))
if __name__ == "__main__":
  unittest.main(defaultTest="regressionTest")

希望本文所述對大家的Python程序設計有所幫助。

您可能感興趣的文章: