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 unittest

class 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

那如何生成一個測試報告呢，需要加入另外一個模塊了，HTMLTestRunner，這個模塊需要自己安裝，使用執行測試用例就會生成一個html的測試報告，里面會有每個測試用例的執行結果，代碼如下：

import HTMLTestRunner

import unittest

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

def tearDown(self):

#每個測試用例執行之后做操作

print('111')

def setUp(self):

#每個測試用例執行之前做操作

print(22222)

def test_run(self):

# self.assertEqual(1,1)

self.assertIs(1,1)

#測試用例

def test_run2(self):

# self.assertEqual(1,1)

self.assertIs(1,1)

#測試用例

def test_run3(self):

# self.assertEqual(1,1)

self.assertIs(1,1)

#測試用例

def test_run1(self):

# self.assertEqual(1,1)

self.assertIs(1,1)

#測試用例

if __name__ == '__main__':

test_suite = unittest.TestSuite()#創建一個測試集合

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

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

fp = open('res.html','wb')#打開一個保存結果的html文件

runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='api測試報告',description='測試情況')

#生成執行用例的對象

runner.run(test_suite)

#執行測試套件

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

import unittest,HTMLTestRunner

suite = unittest.TestSuite()#創建測試套件

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

#找到某個目錄下所有的以test開頭的Python文件里面的測試用例

for case in all_cases:

suite.addTests(case)#把所有的測試用例添加進來

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

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

runner.run(suite)

#運行測試

我們在后續進行持續集成的時候，要讓代碼自動運行，就會用到Jenkins了，但是上面產生的測試報告都是html格式的，Jenkins不認識，就在Jenkins里面顯示不出來。那咱們就要產生一些Jenkins認識的測試報告，Jenkins認識xml格式的報告，那咱們就產生xml格式的唄，就需要用一個新的模塊，xmlrunner，安裝直接 pip install xmlrunner即可，代碼如下：

import unittest

import xmlrunner

#導入這個模塊

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

runner.run(test_suite)

然后咱們運行，可以看到在report目錄下已經產生了xml格式的報告了，而且還自動把日期加上了

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。