快捷導(dǎo)航

分享unittest單元測(cè)試框架中幾種常用的用例加載方法

更新時(shí)間：2020年12月02日 15:50:03 作者：小公瑾

這篇文章主要介紹了unittest單元測(cè)試框架中常用的幾種用例加載方法，幫助大家更好的理解和使用python的unittest測(cè)試模塊，感興趣的朋友可以了解下

　　unittest模塊是Python自帶的一個(gè)單元測(cè)試模塊，我們可以用來(lái)做單元測(cè)試。unittest模塊包含了如下幾個(gè)子模塊：

測(cè)試用例：TestCase
測(cè)試集：TestSuite
加載用例：TestLoader
執(zhí)行用例：TextTestRunner

　　首先編寫(xiě)一個(gè)簡(jiǎn)單的加減乘除數(shù)學(xué)方法類(lèi)：

class MathCalculate:
 '''
 加減乘除的計(jì)算類(lèi)
 '''
 def __init__(self, first_num, second_num):
  self.first_num = first_num
  self.second_num = second_num

 def add(self):
  '''
  加法
  :return:
  '''
  return self.first_num + self.second_num

 def subtract(self):
  '''
  減法
  :return:
  '''
  return self.first_num - self.second_num

 def multiply(self):
  '''
  乘法
  :return:
  '''
  return self.first_num * self.second_num

 def divide(self):
  '''
  除法
  :return:
  '''
  try:
   return round(self.first_num / self.second_num, 2)
  except ZeroDivisionError:
   return '∞'

　　接下來(lái)我們利用unittest中的TestCase類(lèi)來(lái)編寫(xiě)測(cè)試用例，編寫(xiě)之前，要先知道TestCase類(lèi)以及類(lèi)方法的用法：

要引用TestCase類(lèi)，需要引入unittest模塊
引用TestCase是編寫(xiě)一個(gè)測(cè)試類(lèi)，所以要新建一個(gè)類(lèi)
測(cè)試類(lèi)里面的方法名都是以test_開(kāi)頭，這樣unittest才能識(shí)別到用例并去執(zhí)行

　　新建一個(gè)模塊，編寫(xiě)一個(gè)簡(jiǎn)單的測(cè)試用例類(lèi)：

import unittest
from python_for_unittest_testsuite.MathCalculate import MathCalculate as MC


class TestMathCalculate(unittest.TestCase):
 '''
 測(cè)試加減乘除類(lèi)
 '''

 def test_add(self):
  actual = MC(1, 1).add()
  self.assertEqual(2, actual, msg='兩數(shù)相加測(cè)試失敗')

 def test_subtract(self):
  actual = MC(1, 1).subtract()
  self.assertEqual(0, actual, msg='兩數(shù)相減測(cè)試失敗')

 def test_multiply(self):
  actual = MC(1, 1).multiply()
  self.assertEqual(1, actual, msg='兩數(shù)相乘測(cè)試失敗')

 def test_divide(self):
  actual = MC(1, 1).divide()
  self.assertEqual(1, actual, msg='兩數(shù)相除測(cè)試失敗')


if __name__ == '__main__':
 unittest.main()

　　我們運(yùn)行代碼之后，得到如下結(jié)果：

　　但是這樣遠(yuǎn)遠(yuǎn)達(dá)不到我們的要求，還需要繼續(xù)優(yōu)化，接下來(lái)進(jìn)入正題......

　　上面我們看到的是利用unittest.main()執(zhí)行所有的用例，但是有的時(shí)候我們不需要執(zhí)行所有的測(cè)試用例，并且可以按照自己的需求去添加用例，這時(shí)我們就要使用TestSuite來(lái)幫助我們更好地去實(shí)現(xiàn)單元測(cè)試

　　方法一：加載測(cè)試用例，直接使用addTest()方法添加測(cè)試用例，代碼示范如下：

import unittest
from python_for_unittest_testsuite.test_math_calculate import TestMathCalculate

suite = unittest.TestSuite()
# 增加一條用例 addTest()方法傳入：類(lèi)名(測(cè)試方法名)
suite.addTest(TestMathCalculate('test_add'))
suite.addTest(TestMathCalculate('test_subtract'))
suite.addTest(TestMathCalculate('test_multiply'))
suite.addTest(TestMathCalculate('test_divide'))

# 上下文管理器，生成測(cè)試報(bào)告
with open('test_math.txt', mode='w+', encoding='utf-8') as f:
 runner = unittest.TextTestRunner(f, 'text', 2)
 runner.run(suite)

　　方法二：加載測(cè)試類(lèi)，通過(guò)TestLoader里面的loadTestsFromTestCase來(lái)加載測(cè)試類(lèi)，代碼示范如下：

import unittest
from python_for_unittest_testsuite.test_math_calculate import TestMathCalculate

suite = unittest.TestSuite()
loader = unittest.TestLoader()
# 加載測(cè)試類(lèi)
suite.addTest(loader.loadTestsFromTestCase(TestMathCalculate))


# 上下文管理器，生成測(cè)試報(bào)告
with open('test_math.txt', mode='w+', encoding='utf-8') as f:
 runner = unittest.TextTestRunner(f, 'text', 2)
 runner.run(suite)

　　方法三：加載測(cè)試模塊，通過(guò)TestLoader里面的loadTestsFromModule來(lái)加載測(cè)試類(lèi)，代碼示范如下：

import unittest
from python_for_unittest_testsuite import test_math_calculate


suite = unittest.TestSuite()
loader = unittest.TestLoader()
# 加載測(cè)試模塊
suite.addTest(loader.loadTestsFromModule(test_math_calculate))


# 上下文管理器，生成測(cè)試報(bào)告
with open('test_math.txt', mode='w+', encoding='utf-8') as f:
 runner = unittest.TextTestRunner(f, 'text', 2)
 runner.run(suite)

　　方法四：自動(dòng)搜尋測(cè)試用例，根據(jù)傳入的測(cè)試模塊的所屬目錄路徑自動(dòng)去查找測(cè)試用例，代碼示范如下：

import unittest

# 根據(jù)傳入的測(cè)試模塊的所屬目錄路徑自動(dòng)去查找
suite = unittest.defaultTestLoader.discover("G:\pycharm\PycharmProjects\python_class\python_for_unittest_testsuite")

# 上下文管理器，生成測(cè)試報(bào)告
with open('test_math.txt', mode='w+', encoding='utf-8') as f:
 runner = unittest.TextTestRunner(f, 'text', 2)
 runner.run(suite)

以上就是unittest單元測(cè)試框架中常用的幾種用例加載方法的詳細(xì)內(nèi)容，更多關(guān)于unittest單元測(cè)試框架用例加載的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: