Python中的Unittest基本使用

更新時間：2022年08月26日 11:41:01 作者：哲思天下

這篇文章主要介紹了Python中的Unittest基本使用，文章圍繞主題展開詳細的內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下

前言：

Unittest又名單元測試，主要用于測試自己寫的代碼的正確性和完備性，也方便自己和他人對代碼的維護。在軟件開發(fā)過程中，測試驅動開發(fā)的理念是一種好的開發(fā)習慣。

例如現(xiàn)在我們寫了一個求絕對值的函數(shù)my_abs，存放在myAbs.py中

def my_abs(number):
    if not (isinstance(number, int) or isinstance(number, float)):
        raise ValueError("invalid value %s" % number)
    else:
        if(number >= 0):
            return number
        else:
            return -number

此處的my_abs函數(shù)對參數(shù)做了檢查，參數(shù)必須是一個數(shù)（python的數(shù)只有int和float型）。那么如果想要測試這個函數(shù)的正確性，我們需要寫一個測試函數(shù)，用于測試函數(shù)的各個分支情況。

基于unittest的的測試代碼如下：

import unittest
 
from myAbs import my_abs
class TestmyAbs(unittest.TestCase):
    def test_positive(self):
        self.assertEqual(my_abs(1), 1)
 
    def test_zero(self):
        self.assertEqual(my_abs(0), 0)
 
    def test_negative(self):
        self.assertEqual(my_abs(-1), 1)
 
    def test_inputerror(self):
        with self.assertRaises(ValueError):
            my_abs("1")
 
if __name__ == "__main__":
    unittest.main()

上述代碼存放在test_myAbs.py中。

unittest的基本使用方法是：

1. 文件命名以test_開頭，這樣方便unittest自動執(zhí)行，unittest能自動識別目錄下test_開頭的文件執(zhí)行
2. import unittest 這個是運行unittest的前提
3. 定義class要以Test開頭，必須要從unittest.TestCase繼承，且不能有init方法，測試的方法或者函數(shù)以test_開頭
4. 測試方法中使用assert進行斷言

在上面測試代碼的TestMyabs類的方法中，分別對正數(shù)，零和負數(shù)的情況進行了測試，使用assertEqual進行結果的判定；另外測試中還對輸入?yún)?shù)是無效的情況進行了判定，使用了with語句和assertRaises的方式，針對期望拋出來的異常進行判定

unittest的運行方式：

1. 單獨運行test_myAbs.py文件，前提是在文件尾增加

? ??if __name__ == "__main__":
? ? ? ? unittest.main()

2. 發(fā)送命令 python -m unittest .\myAbs.py

當發(fā)送命令python -m unittest時，會執(zhí)行當前目錄下所有test_開頭的文件

運行結果如下：

Testing started at 4:43 PM ...
Launching pytest with arguments C:/work/myPythonPrj/test_myAbs.py --no-header --no-summary -q in C:\work\myPythonPrj
 
============================= test session starts =============================
collecting ... collected 4 items
 
test_myAbs.py::TestmyAbs::test_inputerror PASSED                         [ 25%]
test_myAbs.py::TestmyAbs::test_negative PASSED                           [ 50%]
test_myAbs.py::TestmyAbs::test_positive PASSED                           [ 75%]
test_myAbs.py::TestmyAbs::test_zero PASSED                               [100%]
 
============================== 4 passed in 0.03s ==============================
 
Process finished with exit code 0

到此這篇關于Python中的Unittest基本使用的文章就介紹到這了,更多相關Python Unittest 內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: