單元測試是用來對一個模塊、一個函數(shù)或者一個類來進行正確性檢驗的測試工作矛物。
比如對函數(shù)abs()茫死,我們可以編寫出以下幾個測試用例:
輸入正數(shù),比如1履羞、1.2峦萎、0.99,期待返回值與輸入相同吧雹;
輸入負數(shù),比如-1涂身、-1.2雄卷、-0.99,期待返回值與輸入相反蛤售;
輸入0丁鹉,期待返回0;
輸入非數(shù)值類型悴能,比如None揣钦、[]、{}漠酿,期待拋出TypeError冯凹。
把上面的測試用例放到一個測試模塊里,就是一個完整的單元測試炒嘲。
如果單元測試通過宇姚,說明我們測試的這個函數(shù)能夠正常工作匈庭。如果單元測試不通過,要么函數(shù)有bug浑劳,要么測試條件輸入不正確阱持,總之,需要修復使單元測試能夠通過魔熏。
單元測試通過后有什么意義呢衷咽?如果我們對abs()函數(shù)代碼做了修改,只需要再跑一遍單元測試蒜绽,如果通過镶骗,說明我們的修改不會對abs()函數(shù)原有的行為造成影響,如果測試不通過滓窍,說明我們的修改與原有行為不一致卖词,要么修改代碼,要么修改測試吏夯。
這種以測試為驅(qū)動的開發(fā)模式最大的好處就是確保一個程序模塊的行為符合我們設(shè)計的測試用例此蜈。在將來修改的時候,可以極大程度地保證該模塊行為仍然是正確的噪生。
我們來編寫一個Dict類裆赵,這個類的行為和dict一致,但是可以通過屬性來訪問跺嗽,mydict.py代碼如下:
為了編寫單元測試,我們需要引入Python自帶的unittest模塊植兰,編寫mydict_test.py如下:
編寫單元測試時楣导,我們需要編寫一個測試類畜挨,從unittest.TestCase繼承。
以test開頭的方法就是測試方法毡咏,不以test開頭的方法不被認為是測試方法逮刨,測試的時候不會被執(zhí)行。
對每一類測試都需要編寫一個test_xxx()方法。由于unittest.TestCase提供了很多內(nèi)置的條件判斷落恼,我們只需要調(diào)用這些方法就可以斷言輸出是否是我們所期望的离熏。最常用的斷言就是assertEqual():
self.assertEqual(abs(-1), 1) # 斷言函數(shù)返回的結(jié)果與1相等
另一種重要的斷言就是期待拋出指定類型的Error滋戳,比如通過d[’empty’]訪問不存在的key時,斷言會拋出KeyError:
with self.assertRaises(KeyError):
? ? value = d['empty']
而通過d.empty訪問不存在的key時奸鸯,我們期待拋出AttributeError:
with self.assertRaises(AttributeError):
? ? value = d.empty
運行單元測試
一旦編寫好單元測試娄涩,我們就可以運行單元測試。最簡單的運行方式是在mydict_test.py的最后加上兩行代碼:
if __name__ == '__main__':
? ? unittest.main()
這樣就可以把mydict_test.py當做正常的python腳本運行:
$ python mydict_test.py
另一種方法是在命令行通過參數(shù)-m unittest直接運行單元測試:
$ python -m unittest mydict_test
.....
----------------------------------------------------------------------
Ran 5 tests in 0.000s
OK
這是推薦的做法扬虚,因為這樣可以一次批量運行很多單元測試球恤,并且,有很多工具可以自動來運行這些單元測試堪置。我們也可以在pycharm里面直接運行測試舀锨。
setUp與tearDown
可以在單元測試中編寫兩個特殊的setUp()和tearDown()方法宛逗。這兩個方法會分別在每調(diào)用一個測試方法的前后分別被執(zhí)行。
setUp()和tearDown()方法有什么用呢碑诉?設(shè)想你的測試需要啟動一個數(shù)據(jù)庫侥锦,這時恭垦,就可以在setUp()方法中連接數(shù)據(jù)庫,在tearDown()方法中關(guān)閉數(shù)據(jù)庫唠帝,這樣玄柏,不必在每個測試方法中重復相同的代碼:
class TestDict(unittest.TestCase):
? ? def setUp(self):
? ? ? ? print('setUp...')
? ? def tearDown(self):
? ? ? ? print('tearDown...')
小結(jié)
單元測試可以有效地測試某個程序模塊的行為粪摘,是未來重構(gòu)代碼的信心保證。
單元測試的測試用例要覆蓋常用的輸入組合徘意、邊界條件和異常椎咧。
單元測試代碼要非常簡單,如果測試代碼太復雜蟋座,那么測試代碼本身就可能有bug蜈七。
單元測試通過了并不意味著程序就沒有bug了莫矗,但是不通過程序肯定有bug。