從零開始學Python-Day50-單元測試

單元測試是用來對一個模塊、一個函數(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。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末三娩,一起剝皮案震驚了整個濱河市雀监,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌会前,老刑警劉巖匾竿,帶你破解...
    沈念sama閱讀 219,427評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件岭妖,死亡現(xiàn)場離奇詭異反璃,居然都是意外死亡假夺,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,551評論 3 395
  • 文/潘曉璐 我一進店門梧田,熙熙樓的掌柜王于貴愁眉苦臉地迎上來柿扣,“玉大人未状,你說我怎么就攤上這事析桥。” “怎么了泡仗?”我有些...
    開封第一講書人閱讀 165,747評論 0 356
  • 文/不壞的土叔 我叫張陵搔课,是天一觀的道長截亦。 經(jīng)常有香客問我,道長袍啡,這世上最難降的妖魔是什么境输? 我笑而不...
    開封第一講書人閱讀 58,939評論 1 295
  • 正文 為了忘掉前任嗅剖,我火速辦了婚禮嘁扼,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘蒋院。我一直安慰自己欺旧,他們只是感情好蛤签,可當我...
    茶點故事閱讀 67,955評論 6 392
  • 文/花漫 我一把揭開白布震肮。 她就那樣靜靜地躺著,像睡著了一般鲫尊。 火紅的嫁衣襯著肌膚如雪疫向。 梳的紋絲不亂的頭發(fā)上豪嚎,一...
    開封第一講書人閱讀 51,737評論 1 305
  • 那天搔驼,我揣著相機與錄音,去河邊找鬼侈询。 笑死舌涨,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的扔字。 我是一名探鬼主播囊嘉,決...
    沈念sama閱讀 40,448評論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼革为!你這毒婦竟也來了哗伯?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,352評論 0 276
  • 序言:老撾萬榮一對情侶失蹤篷角,失蹤者是張志新(化名)和其女友劉穎焊刹,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體恳蹲,經(jīng)...
    沈念sama閱讀 45,834評論 1 317
  • 正文 獨居荒郊野嶺守林人離奇死亡虐块,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,992評論 3 338
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了贺奠。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片挂据。...
    茶點故事閱讀 40,133評論 1 351
  • 序言:一個原本活蹦亂跳的男人離奇死亡个绍,死狀恐怖巴柿,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情袁波,我是刑警寧澤,帶...
    沈念sama閱讀 35,815評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站夭苗,受9級特大地震影響猾瘸,放射性物質(zhì)發(fā)生泄漏牵触。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,477評論 3 331
  • 文/蒙蒙 一鲤屡、第九天 我趴在偏房一處隱蔽的房頂上張望癌刽。 院中可真熱鬧显拜,春花似錦、人聲如沸譬淳。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,022評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春场晶,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背扳炬。 一陣腳步聲響...
    開封第一講書人閱讀 33,147評論 1 272
  • 我被黑心中介騙來泰國打工缩多, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 48,398評論 3 373
  • 正文 我出身青樓四苇,卻偏偏與公主長得像胎许,于是被迫代替她去往敵國和親罗售。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 45,077評論 2 355