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函數中關閉連接。

聲明：本網頁內容旨在傳播知識，若有侵權等問題請及時與本網聯系，我們將在第一時間刪除處理。TEL:177 7030 7066 E-MAIL:11247931@qq.com