python學習筆記- 單元測試，UnitTest

所謂的單元測試，就是對一個模組，一個函數，或則是一個類別進行正確性檢測的一類測試工作。

以測試驅動的開發方式叫做測試驅動開發(Test Drived Development). 這種開發模式不僅僅是測試，而是分解需求，保證程式碼品質的一種手段。

單元測試不只在測試的時候保障函數、模組或類別的功能符合我們的需求。同時，也有利於我們在未來修改類別或函數的時候，只需要再跑一遍單元測試就可以確保我們的修改對功能不會產生影響。

當然，單元測試也會讓程式碼量大幅增加。

編寫單元測試程式碼需要引入Python的unittest套件。我們可以新建一個單元測試類，但是這個類別需要繼承unitest.TestCase類別。測試類別中的測試方法必須是test開頭的，否則不會執行。那麼這樣我們就可以使用unitest.TestCase中的方法了。常用的方法有：

assertEqual(self, first, second) # 判斷是否為預期值
assertTrue(bool) # 判斷是否為真
assertRaises(exception) #判斷是非拋出特定異常。

我們來舉一個例子：

我們先定義一個類別自訂類別，然後對它進行單元測試。

Dict.py

class Dict(dict):
def __init__(self, **kwargs):
super(Dict, self).__init__(**kwargs)
def __getattr__(self, item):
try:
return self[item]
except KeyError:
raise AttributeError
def __setattr__(self, key, value):
self[key] = value
import unittest
from Dict import Dictclass DictTest(unittest.TestCase):
def test_init(self):
d = Dict(a=100, b=200)
self.assertEqual(d.a, 100)
self.assertEqual(d.b, 200)
self.assertTrue(isinstance(d, dict))
def test_key(self):
d = Dict()
d['key'] = "300"
self.assertEqual(d["key"], "300")
def test_attar(self):
d = Dict()
d.attar = "attar"
self.assertTrue("attar" in d)
self.assertEqual(d.attar, "attar")
def test_keyError(self):
d = Dict()
with self.assertRaises(KeyError):
value = d["NotExisted"]
def test_attarError(self):
d = Dict()
with self.assertRaises(AttributeError):
value = d.not_existed
if __name__ == '__main__':
unittest.main()

因為我們定義了一個dict的子類，所以我們要對它的key、value進行測試；同時它是一個類，所以我們也需要對它的attribute進行測試。

我們定義的unittest.TestCase子類，需要寫上兩句：

if __name__ == '__main__':
unittest.main()

我們才能像普通python檔案一樣執行它。

另外，我們也可以用以下方式執行多個test

python -m unittest DictTest
setUp 和 tearDown函數。

setUp和tearDown函數是兩個特殊的函數。 setUp再所有單元測試開始前運行； tearDown函數在所有單元測試運行後運行。這兩個函數的應用場景主要有：例如在需要連接資料庫的情況下，可以在setUp函數中開始連接，在tearDown函數中關閉連接。

以上就是python學習筆記- 單元測試，UnitTest的內容，更多相關內容請關注PHP中文網（www.php.cn）！