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進(jìn)行斷言

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

unittest的運行方式：

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

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

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

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

運行結(jié)果如下：

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

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

您可能感興趣的文章: