當我們在寫程序的時候，我們需要通過測試來驗證程序是否出錯或者存在問題，但是，編寫大量的測試來確保程序的每個細節都沒問題會顯得很繁瑣。在Python中，我們可以借助一些標準模塊來幫助我們自動完成測試過程，比如：

unittest: 一個通用的測試框架；

doctest: 一個更簡單的模塊，是為檢查文檔而設計的，但也非常適合用來編寫單元測試。

下面，筆者將會簡單介紹這兩個模塊在測試中的應用。

doctest

doctest模塊會搜索那些看起來像是python交互式會話中的代碼片段，然后嘗試執行并驗證結果。下面我們以doctest.testmod為例，函數doctest.testmod會讀取模塊中的所有文檔字符串，查找看起來像是從交互式解釋器中摘取的示例，再檢查這些示例是否反映了實際情況。

我們先創建示例代碼文件test_string_lower.py，完整代碼如下：

# -*- coding: utf-8 -*-

def string_lower(string):

'''

返回一個字符串的小寫

:param string: type: str

:return: the lower of input string

>>> string_lower('AbC')

'abc'

>>> string_lower('ABC')

'abc'

>>> string_lower('abc')

'abc'

'''

return string.lower()

if __name__ == '__main__':

import doctest, test_string_lower

doctest.testmod(test_string_lower)

首先先對程序進行說明，函數string_lower用于返回輸入字符串的小寫，函數中的注釋中，一共包含了3個測試實例，期望盡可能地包含各種測試情況，接著在主函數中導入doctest, test_string_lower，再運行doctest中的testmod函數即可進行測試。

接著，我們開始測試。首先，在命令行中輸入?python test_string_lower.py，運行后會發現什么都沒有輸出，但這其實是件好事，它表明程序中的所有測試都通過了！那么，如果我們想要獲得更多的輸出呢？可在運行腳本的時候增加參數?-v ，這時候命令變成?python test_string_lower.py -v ，輸出的結果如下：

Trying:

string_lower('AbC')

Expecting:

'abc'

ok

Trying:

string_lower('ABC')

Expecting:

'abc'

ok

Trying:

string_lower('abc')

Expecting:

'abc'

ok

1 items had no tests:

test_string_lower

1 items passed all tests:

3 tests in test_string_lower.string_lower

3 tests in 2 items.

3 passed and 0 failed.

Test passed

可以看到，程序測試的背后還是發生了很多事。接著，我們嘗試著程序出錯的情況，比如我們不小心把函數的返回寫成了：

return string.upper()

這其實是返回輸入字符串的大寫了，而我們測試的實例卻返回了輸入字符串的小寫，再運行該腳本(加上參數?-v )，輸出的結果如下：

Failed example:

string_lower('abc')

Expected:

'abc'

Got:

'ABC'

1 items had no tests:

test_string_lower

**********************************************************************

1 items had failures:

3 of 3 in test_string_lower.string_lower

3 tests in 2 items.

0 passed and 3 failed.

***Test Failed*** 3 failures.

這時候，程序測試失敗，它不僅捕捉到了bug，還清楚地指出錯誤出在什么地方。我們不難把這個程序修改過來。

關于doctest模塊的更詳細的使用說明，可以參考網址：?https://docs.python.org/2/lib... 。

unittest

unittest類似于流行的Java測試框架JUnit，它比doctest更靈活，更強大，能夠幫助你以結構化的方式來編寫龐大而詳盡的測試集。

我們以一個簡單的示例入手，首先我們編寫my_math.py腳本，代碼如下：

# -*- coding: utf-8 -*-

def product(x, y):

'''

:param x: int, float

:param y: int, float

:return: x * y

'''

return x * y

該函數實現的功能為：輸入兩個數x, y， 返回這兩個數的乘積。接著是test_my_math.py腳本，完整的代碼如下：

import unittest, my_math

class ProductTestcase(unittest.TestCase):

def setUp(self):

print('begin test')

def test_integers(self):

for x in range(-10, 10):

for y in range(-10, 10):

p = my_math.product(x, y)

self.assertEqual(p, x*y, 'integer multiplication failed')

def test_floats(self):

for x in range(-10, 10):

for y in range(-10, 10):

x = x/10

y = y/10

p = my_math.product(x, y)

self.assertEqual(p, x * y, 'integer multiplication failed')

if __name__ == '__main__':

unittest.main()

函數unittest.main負責替你運行測試：在測試方法前執行setUp方法，示例化所有的TestCase子類，并運行所有名稱以test打頭的方法。assertEqual方法檢車指定的條件(這里是相等)，以判斷指定的測試是成功了還是失敗了。

接著，我們運行前面的測試，輸出的結果如下：

begin test

.begin test

.

----------------------------------------------------------------------

Ran 2 tests in 0.001s

OK

可以看到，該程序運行了兩個測試，每個測試前都會輸出'begin test'，?. 表示測試成功，若測試失敗，則返回的是?F 。

接著模擬測試出錯的情形，將my_math函數中的product方法改成返回：

return x + y

再運行測試腳本，輸出的結果如下：

begin test

Fbegin test

F

======================================================================

FAIL: test_floats (__main__.ProductTestcase)

----------------------------------------------------------------------

Traceback (most recent call last):

File "test_my_math.py", line 20, in test_floats

self.assertEqual(p, x * y, 'integer multiplication failed')

AssertionError: -2.0 != 1.0 : integer multiplication failed

======================================================================

FAIL: test_integers (__main__.ProductTestcase)

----------------------------------------------------------------------

Traceback (most recent call last):

File "test_my_math.py", line 12, in test_integers

self.assertEqual(p, x*y, 'integer multiplication failed')

AssertionError: -20 != 100 : integer multiplication failed

----------------------------------------------------------------------

Ran 2 tests in 0.001s

FAILED (failures=2)

兩條測試都未通過，返回的是?F ，并幫助你指出了錯誤的地方，接下來，你應該能快速地修復這個bug。

關于unittest模塊的更加詳細的說明，可以參考網址：?https://docs.python.org/3/lib... 。

總結

以上所述是小編給大家介紹的Python中的測試工具，希望對大家有所幫助，如果大家有任何疑問歡迎給我留言，小編會技術回復大家的！

總結

以上是生活随笔為你收集整理的python代码测试工具模块_详解Python中的测试工具的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。