doctest：直接寫(xiě)在方法體中，利用了 python 動(dòng)態(tài)語(yǔ)言的特性，書(shū)寫(xiě)方式簡(jiǎn)單明了，前提是項(xiàng)目不要太大，如果項(xiàng)目復(fù)雜代碼量大，再加上寫(xiě)在方法體中的單元測(cè)試，整體就會(huì)看起來(lái)有些臃腫，因此 doctest 適用于代碼量不大的普通項(xiàng)目
unittest：一般寫(xiě)在獨(dú)立文件中，與 spring 的單元測(cè)試有點(diǎn)像，項(xiàng)目結(jié)構(gòu)清晰，不過(guò)簡(jiǎn)單項(xiàng)目中測(cè)試一些小功能的話實(shí)屬dark不必了

二、實(shí)現(xiàn)

先寫(xiě)一個(gè)待測(cè)試的方法

def add(num1, num2):
    return num1 + num2

1.doctest

def add(num1, num2):
    """
    直接寫(xiě)在待測(cè)方法中
    
    >>> add(14, 5)
    19
    >>> add(14, 5) > 1
    True
    
    >>> add(20, 30, 70)
    Traceback (most recent call last):
        ...
    TypeError: add() takes 2 positional arguments but 3 were given
    """
    return num1 + num2
if __name__ == "__main__":
    import doctest
    doctest.testmod()

運(yùn)行上面的代碼什么都不會(huì)輸出，因?yàn)轭A(yù)測(cè)結(jié)果全部正確，只有在預(yù)測(cè)結(jié)果錯(cuò)誤的時(shí)候才會(huì)給提示
ps：

add(14, 5) 為測(cè)試方法，19 為預(yù)測(cè)結(jié)果
add(14, 5) > 1 為測(cè)試方法，True 為預(yù)測(cè)結(jié)果
add(20, 30, 70) 為測(cè)試方法，Traceback (most recent call last): 為異常拋出的第一句話【錯(cuò)誤原因】，TypeError: add() takes 2 positional arguments but 3 were given 為異常拋出的最后一句話【異常類型：解釋說(shuō)明】，中間三個(gè)點(diǎn)省略了異常中的文件路徑信息

2.unittest

import unittest
# 引入上面的待測(cè)試方法
from src.main.add import add
class TestStatisticalFunctions(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(14, 5), 19)            # 斷言 add(14, 5) 的結(jié)果為 19
        self.assertTrue(add(14, 5) > 1, True)       # 斷言 add(14, 5) > 1 的結(jié)果為 True
        with self.assertRaises(TypeError):          # 斷言 add(20, 30, 70) 拋出 TypeError 異常
            add(20, 30, 70)
if __name__ == '__main__':
    unittest.main()

同樣，運(yùn)行上面的代碼什么都不會(huì)輸出，因?yàn)轭A(yù)測(cè)結(jié)果全部正確，只有在預(yù)測(cè)結(jié)果錯(cuò)誤的時(shí)候才會(huì)給提示

到此這篇關(guān)于Python單元測(cè)試的兩種寫(xiě)法的文章就介紹到這了,更多相關(guān)Python單元測(cè)試內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: