溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
怎么正確的使用unittest單元測試框架？針對這個問題，這篇文章詳細介紹了相對應的分析和解答，希望可以幫助更多想解決這個問題的小伙伴找到更簡單易行的方法。
unittest模塊是Python自帶的一個單元測試模塊，我們可以用來做單元測試。unittest模塊包含了如下幾個子模塊：
首先編寫一個簡單的加減乘除數學方法類：
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類以及類方法的用法：
新建一個模塊，編寫一個簡單的測試用例類：
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)關于怎么正確的使用unittest單元測試框架問題的解答就分享到這里了，希望以上內容可以對大家有一定的幫助，如果你還有很多疑惑沒有解開，可以關注億速云行業資訊頻道了解更多相關知識。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
