關於unittest單元測試框架中常用的幾種用例加載方法


  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類以及類方法的用法:

  1. 要引用TestCase類,需要引入unittest模塊
  2. 引用TestCase是編寫一個測試類,所以要新建一個類
  3. 測試類里面的方法名都是以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:\pycharm\PycharmProjects\python_class\python_for_unittest_testsuite")

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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM