Python中有一個(gè)自帶的單元測(cè)試框架是unittest模塊，用它來(lái)做單元測(cè)試，它里面封裝好了一些校驗(yàn)返回的結(jié)果方法和一些用例執(zhí)行前的初始化操作。

unittest是python的標(biāo)準(zhǔn)測(cè)試庫(kù)，相比于其他測(cè)試框架是python目前使用最廣的測(cè)試框架。

unittest有四個(gè)比較重要的概念是test fixture, test case, test suite, test runner, 。

在說(shuō)unittest之前，先說(shuō)幾個(gè)概念：

TestCase 也就是測(cè)試用例

TestSuite 多個(gè)測(cè)試用例集合在一起，就是TestSuite

TestLoader是用來(lái)加載TestCase到TestSuite中的

TestRunner是來(lái)執(zhí)行測(cè)試用例的,測(cè)試的結(jié)果會(huì)保存到TestResult實(shí)例中，包括運(yùn)行了多少測(cè)試用例，成功了多少，失敗了多少等信息

通過(guò)dir(unittest)，我們可以看到unittest全部的屬性和方法，他們的關(guān)系如下圖所示。

unittest主要類(lèi)關(guān)系

正常調(diào)用unittest的流程是，TestLoader 自動(dòng)將測(cè)試用例TestCase中加載到TestSuite里，TextTestRunner調(diào)用TestSuite的run方法，順序執(zhí)行里面的TestCase中以test開(kāi)頭的方法，并得到測(cè)試結(jié)果TestResult。在執(zhí)行TestCase過(guò)程中，先進(jìn)行SetUp()環(huán)境準(zhǔn)備，執(zhí)行測(cè)試代碼，最后tearDown()進(jìn)行測(cè)試的還原。

其中TestLoader在加載過(guò)程中，進(jìn)行添加的TestCase是沒(méi)有順序的。一個(gè)TestCase里如果存在多個(gè)驗(yàn)證方法的話(huà)，會(huì)按照方法中test后方首字母的ascii碼從小到大排序后執(zhí)行。

可以通過(guò)手動(dòng)調(diào)用TestSuite的addTest、addTests方法來(lái)動(dòng)態(tài)添加TestCase，這樣既可以確定添加用例的執(zhí)行順序，也可避免TestCase中的驗(yàn)證方法一定要用test開(kāi)頭。

下面寫(xiě)一個(gè)簡(jiǎn)單的單元測(cè)試用例

import unittest

class MyTest(unittest.TestCase): # 繼承unittest.TestCase

def tearDown(self):

# 每個(gè)測(cè)試用例執(zhí)行之后做操作

print('111')

def setUp(self):

# 每個(gè)測(cè)試用例執(zhí)行之前做操作

print('22222')

@classmethod

def tearDownClass(self):

# 必須使用 @ classmethod裝飾器, 所有test運(yùn)行完后運(yùn)行一次

print('4444444')

@classmethod

def setUpClass(self):

# 必須使用@classmethod 裝飾器,所有test運(yùn)行前運(yùn)行一次

print('33333')

def test_a_run(self):

self.assertEqual(1, 1) # 測(cè)試用例

def test_b_run(self):

self.assertEqual(2, 2) # 測(cè)試用例

if __name__ == '__main__':

unittest.main()#運(yùn)行所有的測(cè)試用例

下面是一些常用的斷言，也就是校驗(yàn)結(jié)果

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

那如何生成一個(gè)測(cè)試報(bào)告呢，需要加入另外一個(gè)模塊了，HTMLTestRunner，這個(gè)模塊需要自己安裝，使用執(zhí)行測(cè)試用例就會(huì)生成一個(gè)html的測(cè)試報(bào)告，里面會(huì)有每個(gè)測(cè)試用例的執(zhí)行結(jié)果，代碼如下：

import HTMLTestRunner

import unittest

class MyTest(unittest.TestCase):#繼承unittest.TestCase

def tearDown(self):

#每個(gè)測(cè)試用例執(zhí)行之后做操作

print('111')

def setUp(self):

#每個(gè)測(cè)試用例執(zhí)行之前做操作

print(22222)

def test_run(self):

# self.assertEqual(1,1)

self.assertIs(1,1)

#測(cè)試用例

def test_run2(self):

# self.assertEqual(1,1)

self.assertIs(1,1)

#測(cè)試用例

def test_run3(self):

# self.assertEqual(1,1)

self.assertIs(1,1)

#測(cè)試用例

def test_run1(self):

# self.assertEqual(1,1)

self.assertIs(1,1)

#測(cè)試用例

if __name__ == '__main__':

test_suite = unittest.TestSuite()#創(chuàng)建一個(gè)測(cè)試集合

test_suite.addTest(MyTest('test_run1'))#測(cè)試套件中添加測(cè)試用例

#test_suite.addTest(unittest.makeSuite(MyTest))#使用makeSuite方法添加所有的測(cè)試方法

fp = open('res.html','wb')#打開(kāi)一個(gè)保存結(jié)果的html文件

runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='api測(cè)試報(bào)告',description='測(cè)試情況')

#生成執(zhí)行用例的對(duì)象

runner.run(test_suite)

#執(zhí)行測(cè)試套件

如果我們有很多個(gè)模塊，每個(gè)模塊下面都寫(xiě)了很多python文件，每個(gè)python文件里面都有測(cè)試用例，那怎么把這個(gè)目錄下的用例都執(zhí)行了呢，就要先找到這個(gè)目錄下的所有python文件，然后找到里面的測(cè)試用例，逐個(gè)執(zhí)行，代碼如下：

import unittest,HTMLTestRunner

suite = unittest.TestSuite()#創(chuàng)建測(cè)試套件

all_cases = unittest.defaultTestLoader.discover('.','test_*.py')

#找到某個(gè)目錄下所有的以test開(kāi)頭的Python文件里面的測(cè)試用例

for case in all_cases:

suite.addTests(case)#把所有的測(cè)試用例添加進(jìn)來(lái)

fp = open('res.html','wb')

runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='all_tests',description='所有測(cè)試情況')

runner.run(suite)

#運(yùn)行測(cè)試

我們?cè)诤罄m(xù)進(jìn)行持續(xù)集成的時(shí)候，要讓代碼自動(dòng)運(yùn)行，就會(huì)用到Jenkins了，但是上面產(chǎn)生的測(cè)試報(bào)告都是html格式的，Jenkins不認(rèn)識(shí)，就在Jenkins里面顯示不出來(lái)。那咱們就要產(chǎn)生一些Jenkins認(rèn)識(shí)的測(cè)試報(bào)告，Jenkins認(rèn)識(shí)xml格式的報(bào)告，那咱們就產(chǎn)生xml格式的唄，就需要用一個(gè)新的模塊，xmlrunner，安裝直接 pip install xmlrunner即可，代碼如下：

import unittest

import xmlrunner

#導(dǎo)入這個(gè)模塊

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')#指定報(bào)告放的目錄

runner.run(test_suite)

然后咱們運(yùn)行，可以看到在report目錄下已經(jīng)產(chǎn)生了xml格式的報(bào)告了，而且還自動(dòng)把日期加上了

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

總結(jié)

以上是生活随笔為你收集整理的python测试用例怎么写_Python单元测试unittest的具体使用示例的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。