測試
測試用例
對函數(shù) abs（），這個函數(shù)的作用就是取絕對值掺冠，我們可以編寫以下幾個測試用例：
- 輸入正數(shù)，比如 1码党、 1.5德崭、 0.99，期待返回值與輸入相同
- 輸入負數(shù)揖盘，比如 -1眉厨、 -1.5、 -0.99扣讼， 期待返回值與輸入值相反
- 輸入0缺猛， 期待返回0缨叫；
- 輸入非數(shù)值類型椭符，比如 None荔燎、 []、 {}销钝， 期待拋出 TypeError
把上面的測試用例有咨，放到測試模塊中，就是一個完整的單元測試蒸健。
單元測試通過說明我們的函數(shù)能夠正常工作座享，要是不過，就說明函數(shù)還有bug似忧，
那么就得修改渣叛，直到單元測試通過。
單元測試得意義
- 一旦測試通過盯捌，以后的修改不會對abs（）有影響淳衙，如果造成影響，測試就不能通過
- 單元測試饺著，在重構中箫攀，也是經(jīng)常用到的，有了單元測試幼衰，就可以放心的重構
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 如下：
import unittest
from learn.two.測試.mydict import Dict
class TestDict(unittest.TestCase):
def test_init(self):
d = Dict(a = 1, b = '測試')
self.assertEqual(d.a, 1)
self.assertEqual(d.b, '測試')
self.assertTrue(isinstance(d, dict))
# def test_key(self): #這部分是測試用例通不過的例子
# d = Dict()
# d['key'] = 'value'
# self.assertSetEqual(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
if __name__ == '__main__':
unittest.main()
直接運行 mydict_test.py
Testing started at 22:54 ...
E:\py\venv\Scripts\python.exe "D:\JetBrains\PyCharm 2018.1.1\helpers\pycharm\_jb_unittest_runner.py" --target mydict_test.TestDict
Launching unittests with arguments python -m unittest mydict_test.TestDict in E:\pyplace\learn_python3\learn\two\測試
Ran 4 tests in 0.012s
OK
Process finished with exit code 0
以上這就說明單元測試通過了
下面是測試不通過的示范（將注釋的部分放開）：
import unittest
from learn.two.測試.mydict import Dict
class TestDict(unittest.TestCase):
def test_init(self):
d = Dict(a = 1, b = '測試')
self.assertEqual(d.a, 1)
self.assertEqual(d.b, '測試')
self.assertTrue(isinstance(d, dict))
def test_key(self): #這部分是測試用例通不過的例子
d = Dict()
d['key'] = 'value'
self.assertSetEqual(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
if __name__ == '__main__':
unittest.main()
運行 mydict_test.py 結(jié)果：
Testing started at 22:58 ...
E:\py\venv\Scripts\python.exe "D:\JetBrains\PyCharm 2018.1.1\helpers\pycharm\_jb_unittest_runner.py" --target mydict_test.TestDict
Launching unittests with arguments python -m unittest mydict_test.TestDict in E:\pyplace\learn_python3\learn\two\測試
Ran 5 tests in 0.027s
FAILED (failures=1)
Failure
Traceback (most recent call last):
File "D:\python\Python36\lib\unittest\case.py", line 1055, in assertSetEqual
difference1 = set1.difference(set2)
AttributeError: 'str' object has no attribute 'difference'
File "D:\python\Python36\lib\unittest\case.py", line 670, in fail
raise self.failureException(msg)
AssertionError: first argument does not support set difference: 'str' object has no attribute 'difference'
可以看到昔善，控制臺輸入了紅色的錯誤日志朴摊，意味著單元測試不通過
單元測試的寫法
- 需要編寫一個測試類恰聘，從 unittest.TestCase 繼承
- 以 test 開頭的就是測試方法吠式，不是 test 開頭的不被認為是測試方法评姨，測試的時候不被執(zhí)行
- 每個類測試都需要編寫 test_xxx() 方法奴曙，由于 unittest.TestCase 提供了很多內(nèi)置的條件判斷奶甘，我們只需要調(diào)用就可以了
常用方法
- 常用的斷言就是 assertEqual();
self.assertEqual(abs(-1), 1) # 斷言返回的結(jié)果與1相等
- 另一種是拋出指定類型的錯誤赛蔫，比如 d['empty']訪問不到存在的Key時裤唠，就拋出 KeyError
with self.assertRaises(KeyError):
value = d['empty']
如果通過 d.empty 訪問不存在的 key 時挤牛，我們期待拋出 AttributeError：
with self.assertRaises(AttributeError):
value = d.empty
運行單元測試
- 在編寫好的單元測試中，添加以下兩行代碼种蘸，我們就可以運行單元測試
if __name__ == '__main___':
unittest.main()
- 這樣墓赴，我們就可以把 mydict_test.py 當作正常的 Python 腳本運行：
$ python mydict_test.py
總結(jié)
- 寫被測試的類
- 繼承 unittest.TestCase 寫單元測試類
- 通過會顯示綠色 Tests passed，并在輸出日志中顯示 OK
- 不通過顯示紅色 Tests failed航瞭，并在輸出日志中顯示 FAILD
