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

二、實現(xiàn)

先寫一個待測試的方法

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

1.doctest

def add(num1, num2):
    """
    直接寫在待測方法中
    
    >>> 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ù)測結(jié)果全部正確，只有在預(yù)測結(jié)果錯誤的時候才會給提示
ps：

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

2.unittest

import unittest
# 引入上面的待測試方法
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ù)測結(jié)果全部正確，只有在預(yù)測結(jié)果錯誤的時候才會給提示

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

您可能感興趣的文章: