分享unittest單元測試框架中幾種常用的用例載入方法

2020-12-02 18:11:31

　　unittest模組是Python自帶的一個單元測試模組，我們可以用來做單元測試。unittest模組包含了如下幾個子模組：

測試用例：TestCase
測試集：TestSuite
載入用例：TestLoader
執行用例：TextTestRunner

　　首先編寫一個簡單的加減乘除數學方法類：

class MathCalculate:
 '''
 加減乘除的計算類
 '''
 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 '∞'

　　接下來我們利用unittest中的TestCase類來編寫測試用例，編寫之前，要先知道TestCase類以及類方法的用法：

要參照TestCase類，需要引入unittest模組
參照TestCase是編寫一個測試類，所以要新建一個類
測試類裡面的方法名都是以test_開頭，這樣unittest才能識別到用例並去執行

　　新建一個模組，編寫一個簡單的測試用例類：

import unittest
from python_for_unittest_testsuite.MathCalculate import MathCalculate as MC


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

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

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

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

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


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

　　我們執行程式碼之後，得到如下結果：

　　但是這樣遠遠達不到我們的要求，還需要繼續優化，接下來進入正題......

　　上面我們看到的是利用unittest.main()執行所有的用例，但是有的時候我們不需要執行所有的測試用例，並且可以按照自己的需求去新增用例，這時我們就要使用TestSuite來幫助我們更好地去實現單元測試

　　方法一：載入測試用例，直接使用addTest()方法新增測試用例，程式碼示範如下：

import unittest
from python_for_unittest_testsuite.test_math_calculate import TestMathCalculate

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

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

　　方法二：載入測試類，通過TestLoader裡面的loadTestsFromTestCase來載入測試類，程式碼示範如下：

import unittest
from python_for_unittest_testsuite.test_math_calculate import TestMathCalculate

suite = unittest.TestSuite()
loader = unittest.TestLoader()
# 載入測試類
suite.addTest(loader.loadTestsFromTestCase(TestMathCalculate))


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

　　方法三：載入測試模組，通過TestLoader裡面的loadTestsFromModule來載入測試類，程式碼示範如下：

import unittest
from python_for_unittest_testsuite import test_math_calculate


suite = unittest.TestSuite()
loader = unittest.TestLoader()
# 載入測試模組
suite.addTest(loader.loadTestsFromModule(test_math_calculate))


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

　　方法四：自動搜尋測試用例，根據傳入的測試模組的所屬目錄路徑自動去查詢測試用例，程式碼示範如下：

import unittest

# 根據傳入的測試模組的所屬目錄路徑自動去查詢
suite = unittest.defaultTestLoader.discover("G:pycharmPycharmProjectspython_classpython_for_unittest_testsuite")

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

以上就是unittest單元測試框架中常用的幾種用例載入方法的詳細內容，更多關於unittest單元測試框架用例載入的資料請關注it145.com其它相關文章！

分享unittest單元測試框架中幾種常用的用例載入方法

熱門文章