python寫(xiě)單元大多數(shù)都會(huì)用到unittest和mock，測(cè)試代碼覆蓋率都會(huì)用到coverage，最后再用nose把所有的東西都串起來(lái)，這樣每次出版本，都能把整個(gè)項(xiàng)目的單元測(cè)試都運(yùn)行一遍。

Unittest

unittest就不詳細(xì)介紹了，注意幾點(diǎn)：

測(cè)試類(lèi)繼承unittest.TestCase

測(cè)試類(lèi)、測(cè)試方法名字最好以test開(kāi)頭，很多工具能根據(jù)名字來(lái)自動(dòng)運(yùn)行，很方便

測(cè)試類(lèi)里面的setUp/tearDown會(huì)在每個(gè)case執(zhí)行之前/之后執(zhí)行，setUpClass/tearDownClass加上@classmethod在整個(gè)測(cè)試類(lèi)開(kāi)始和結(jié)束的時(shí)候執(zhí)行

測(cè)試文件的main函數(shù)里面加上unittest.main()，就可以直接用python命令運(yùn)行了

Mock

單元測(cè)試?yán)锩姹容^精髓的就是mock了，介紹幾種常見(jiàn)的場(chǎng)景：

1.Mock一個(gè)函數(shù)。其實(shí)有好幾種方法，個(gè)人比較推薦下面這種，看上去很清晰：

defmultiple(a,b):

returna*b

classTestProducer(unittest.TestCase):

defsetUp(self):

self.calculator=Calculator()

@mock.patch('multiple')

deftest_multiple(self,mock_multiple):

mock_multiple.return_value=3

self.assertEqual(multiple(8,14),3)

2.Mock一個(gè)對(duì)象里面的方法

classCalculator(object):

defadd(self,a,b):

returna+b

classTestProducer(unittest.TestCase):

defsetUp(self):

self.calculator=Calculator()

@mock.patch.object(Calculator,'add')

deftest_add(self,mock_add):

mock_add.return_value=3

self.assertEqual(self.calculator.add(8,14),3)

3.讓Mock的函數(shù)每次被調(diào)用返回不同的值，而1,2中的方法每次調(diào)用都會(huì)返回同樣的值

classTestProducer(unittest.TestCase):

@mock.patch.object(Calculator,'add')

deftest_effect(self,mock_add):

mock_add.side_effect=[1,2,3]

self.assertEqual(self.calculator.add(8,14),1)

self.assertEqual(self.calculator.add(8,14),2)

self.assertEqual(self.calculator.add(8,14),3)

4.讓Mock的函數(shù)拋出exception

defis_error(self):

try:

os.mkdir("11")

returnFalse

exceptExceptionase:

returnTrue

classTestProducer(unittest.TestCase):

@mock.patch('os.mkdir')

deftest_exception(self,mock_mkdir):

mock_mkdir.side_effect=Exception

self.assertEqual(self.calculator.is_error(),True)

5.Mock多個(gè)函數(shù)，主要是注意順序

@mock.patch.object(Calculator,'add')

@mock.patch('test_unit.multiple')

deftest_both(self,mock_multiple,mock_add):

mock_add.return_value=1

mock_multiple.return_value=2

self.assertEqual(self.calculator.add(8,14),1)

self.assertEqual(multiple(8,14),2)

Coverage

打命令coverage加測(cè)試文件，就可以得到覆蓋率，可以生成html格式的報(bào)告，每次運(yùn)行一個(gè)文件都會(huì)生成一個(gè).coverage文件，需要將combine所有結(jié)果才能得到一個(gè)完整的報(bào)告。

具體的命令參數(shù)參看：http://nedbatchelder.com/code/coverage/cmd.html

更加有用的是配置文件，參看：http://nedbatchelder.com/code/coverage/config.html

配置文件中最有用的功能就是可以不測(cè)某些行的覆蓋率，例如：

[report]

exclude_lines=

#只要在某一行加上注釋“#pragma:nocover”這一行就會(huì)被忽略

pragma:nocover

#忽略掉main函數(shù)

if__name__==.__main__.:

Nose

Nose可以將所有的單元測(cè)試文件一次全部執(zhí)行，并且提供了coverage的插件，能夠統(tǒng)計(jì)整體的覆蓋率。

Nose會(huì)掃描目標(biāo)目錄，如果發(fā)現(xiàn)目錄名以“test”或者“Test”開(kāi)頭，則遞歸地進(jìn)去掃描，并自動(dòng)運(yùn)行所有發(fā)現(xiàn)的以“test”或者“Test”開(kāi)頭的測(cè)試文件。

另外Nose增加了報(bào)級(jí)別的setup和teardown，只需將他們放到__init__.py文件中即可。

Nose命令的執(zhí)行，最簡(jiǎn)單的就是nosetest后面加上你的所有測(cè)試文件或者測(cè)試文件所在的目錄，一些運(yùn)行參數(shù)參看：http://nose.readthedocs.org/en/latest/usage.html

Nose的參數(shù)里面以"--cover"開(kāi)頭的都是coverage相關(guān)的，但是我發(fā)現(xiàn)并沒(méi)有辦法是用coverage的配置文件，需要手動(dòng)安裝一下nose-cov，然后用“--cov-config”來(lái)指定配置文件，其他參數(shù)參看：https://pypi.python.org/pypi/nose-cov

我的項(xiàng)目因?yàn)闇y(cè)試文件比分散，并且有些并沒(méi)有以test開(kāi)頭，所以比較麻煩，只能寫(xiě)了一個(gè)腳本，把這些都串起來(lái)：

importos

importsubprocess

######################################################################

#需要測(cè)試覆蓋率的文件或者目錄

cover_list=[

'src/sample/analyzer/unpacker/src/emulator.py',

'src/sample/analyzer/unpacker/src/emulator_manager.py',

'src/sample/analyzer/unpacker/src/unpacker_analyzer.py',

'src/sample/analyzer/bitvalue/src/confparser.py',

'src/sample/analyzer/bitvalue/src/trunk.py',

]

#測(cè)試用例所在的文件或者目錄，如果測(cè)試文件沒(méi)有以test開(kāi)頭，則必須制定文件名

ut_list=[

'src/sample/analyzer/unpacker/ut',

'src/sample/analyzer/bitvalue/ut/ut_main.py'

]

######################################################################

PRODUCTION_HOME=os.environ.get("PRODUCTION_HOME","../..")

defget_command():

command=[

'nosetests',

'--with-cov',

'--cover-erase',

'--cov-report','html',

'--cov-config','cover.config',

]

forcoverincover_list:

command.append('--cov')

command.append(os.path.join(PRODUCTION_HOME,cover))

forutinut_list:

command.append(os.path.join(PRODUCTION_HOME,ut))

returncommand

if__name__=='__main__':

command=get_command()

printcommand

os.chdir(PRODUCTION_HOME)

proc=subprocess.Popen(command,shell=False,stdout=subprocess.PIPE,stderr=subprocess.PIPE)

output,error=proc.communicate()

return_code=proc.poll()

printoutput

printerror

printreturn_code

以上內(nèi)容為大家介紹了Python單元測(cè)試經(jīng)驗(yàn)總結(jié)，希望對(duì)大家有所幫助，如果想要了解更多Python相關(guān)知識(shí)，請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。http://m.2667701.com/