單元測試 · python

[TOC] mydict.py代碼如下： ~~~ class Dict(dict): def __init__(self, **kw): super().__init__(**kw) def __getattr__(self, key): try: return self[key] except KeyError: raise AttributeError(r"'Dict' object has no attribute '%s'" % key) def __setattr__(self, key, value): self[key] = value ~~~ 為了編寫單元測試，我們需要引入Python自帶的unittest模塊，編寫mydict_test.py如下： ~~~ import unittest from mydict import Dict class TestDict(unittest.TestCase): def test_init(self): d = Dict(a=1, b='test') self.assertEqual(d.a, 1) self.assertEqual(d.b, 'test') self.assertTrue(isinstance(d, dict)) def test_key(self): d = Dict() d['key'] = 'value' self.assertEqual(d.key, 'value') def test_attr(self): d = Dict() d.key = 'value' self.assertTrue('key' in d) self.assertEqual(d['key'], 'value') def test_keyerror(self): d = Dict() with self.assertRaises(KeyError): value = d['empty'] def test_attrerror(self): d = Dict() with self.assertRaises(AttributeError): value = d.empty ~~~ 編寫單元測試時，我們需要編寫一個測試類，從unittest.TestCase繼承。以test開頭的方法就是測試方法，不以test開頭的方法不被認為是測試方法，測試的時候不會被執行。對每一類測試都需要編寫一個test_xxx()方法。由于unittest.TestCase提供了很多內置的條件判斷，我們只需要調用這些方法就可以斷言輸出是否是我們所期望的。最常用的斷言就是assertEqual()： ~~~ self.assertEqual(abs(-1), 1) # 斷言函數返回的結果與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 ~~~ 另一種方法是在命令行通過參數-m unittest直接運行單元測試： ~~~ $ python -m unittest mydict_test ..... ---------------------------------------------------------------------- Ran 5 tests in 0.000s OK ~~~ 這是推薦的做法，因為這樣可以一次批量運行很多單元測試，并且，有很多工具可以自動來運行這些單元測試 **setUp與tearDown** 可以在單元測試中編寫兩個特殊的setUp()和tearDown()方法。這兩個方法會分別在每調用一個測試方法的前后分別被執行。 setUp()和tearDown()方法有什么用呢？設想你的測試需要啟動一個數據庫，這時，就可以在setUp()方法中連接數據庫，在tearDown()方法中關閉數據庫，這樣，不必在每個測試方法中重復相同的代碼： ~~~ class TestDict(unittest.TestCase): def setUp(self): print('setUp...') def tearDown(self): print('tearDown...') ~~~ 可以再次運行測試看看每個測試方法調用前后是否會打印出setUp...和tearDown...