Python標準庫包含兩個測試工具。

doctest：一個簡單的模組，為檢查文件而設計，但也適合用來編寫單元測試。

unittest：一個通用的測試框架。

一、使用doctest進行單元測試

建立檔案mymath.py，內容

def square(x):
  '''
  計算平方並返回結果（下面是單元測試的格式）
  >>> square(2)
  >>> square(3)
  '''
  return x * x
if __name__ == '__main__':
  import doctest,mymath
  doctest.testmod(mymath)

在Sublime Text 3中執行只是提示[Finished in 0.2s]。

在cmd命令下切換到mymath.py所在目錄，執行python mymath.py -v後，有提示測試詳細資訊，測試通過。如下圖：

把函式square裡面的return x * x 改成 return x / x。

再次執行python mymath.py -v，提示測試不通過，如下圖：

二、使用unittest進行單元測試

還是用mymath.py測試，內容：

def square(x):  
  return x * x

新建單元測試檔案test_math.py，內容：

import unittest,mymath
class mathTestCase(unittest.TestCase):  
  def test_square(self):    
    self.assertEqual(mymath.square(2),4)
    self.assertEqual(mymath.square(3),9)
if __name__ == '__main__':
  unittest.main()

執行後，顯示OK 測試通過。

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

把函式square裡面的return x * x 改成 return x / x。

再次執行，這次顯示FAILED 測試不通過。

F
======================================================================
FAIL: test_square (__main__.mathTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
 File "D:\projects\python\test_math.py",line 6,in test_square
  self.assertEqual(mymath.square(2),4)
AssertionError: 1.0 != 4

----------------------------------------------------------------------
Ran 1 test in 0.000s

FAILED (failures=1)

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。