| Method | Checks that | New in |
| assertEqual(a, b) | a == b |
|
| assertNotEqual(a, b) | a != b |
|
| assertTrue(x) | bool(x) is True |
|
| assertFalse(x) | bool(x) is False |
|
| assertIs(a, b) | a is b |
2.7 |
| assertIsNot(a, b) | a is not b |
2.7 |
| assertIsNone(x) | x is None |
2.7 |
| assertIsNotNone(x) | x is not None |
2.7 |
| assertIn(a, b) | a in b |
2.7 |
| assertNotIn(a, b) | a not in b |
2.7 |
| assertIsInstance(a, b) | isinstance(a, b) |
2.7 |
| assertNotIsInstance(a, b) | not isinstance(a, b) |
2.7 |
如果我們在 factorial.py 中調(diào)用 div(0)，我們能看到異常被拋出。
我們也能測試這些異常，就像這樣：
self.assertRaises(ZeroDivisionError, div, 0)
完整代碼：
import unittest
from factorial import fact, div
class TestFactorial(unittest.TestCase):
"""
我們的基本測試類
"""
def test_fact(self):
"""
實際測試
任何以 `test_` 開頭的方法都被視作測試用例
"""
res = fact(5)
self.assertEqual(res, 120)
def test_error(self):
"""
測試由運行時錯誤引發(fā)的異常
"""
self.assertRaises(ZeroDivisionError, div, 0)
if __name__ == '__main__':
unittest.main()
mounttab.py 中只有一個 mount_details() 函數(shù)，函數(shù)分析并打印掛載詳細信息。
import os
def mount_details():
"""
打印掛載詳細信息
"""
if os.path.exists('/proc/mounts'):
fd = open('/proc/mounts')
for line in fd:
line = line.strip()
words = line.split()
print('{} on {} type {}'.format(words[0],words[1],words[2]), end=' ')
if len(words) > 5:
print('({})'.format(' '.join(words[3:-2])))
else:
print()
fd.close()
if __name__ == '__main__':
mount_details()
重構(gòu) mounttab.py
現(xiàn)在我們在 mounttab2.py 中重構(gòu)了上面的代碼并且有一個我們能容易的測試的新函數(shù) parse_mounts()。
import os
def parse_mounts():
"""
分析 /proc/mounts 并 返回元祖的列表
"""
result = []
if os.path.exists('/proc/mounts'):
fd = open('/proc/mounts')
for line in fd:
line = line.strip()
words = line.split()
if len(words) > 5:
res = (words[0],words[1],words[2],'({})'.format(' '.join(words[3:-2])))
else:
res = (words[0],words[1],words[2])
result.append(res)
fd.close()
return result
def mount_details():
"""
打印掛載詳細信息
"""
result = parse_mounts()
for line in result:
if len(line) == 4:
print('{} on {} type {} {}'.format(*line))
else:
print('{} on {} type {}'.format(*line))
if __name__ == '__main__':
mount_details()
同樣我們測試代碼，編寫 mounttest.py 文件：
#!/usr/bin/env python
import unittest
from mounttab2 import parse_mounts
class TestMount(unittest.TestCase):
"""
我們的基本測試類
"""
def test_parsemount(self):
"""
實際測試
任何以 `test_` 開頭的方法都被視作測試用例
"""
result = parse_mounts()
self.assertIsInstance(result, list)
self.assertIsInstance(result[0], tuple)
def test_rootext4(self):
"""
測試找出根文件系統(tǒng)
"""
result = parse_mounts()
for line in result:
if line[1] == '/' and line[2] != 'rootfs':
self.assertEqual(line[2], 'ext4')
if __name__ == '__main__':
unittest.main()
運行程序
$ python3 mounttest.py
..
----------------------------------------------------------------------
Ran 2 tests in 0.001s
OK
測試覆蓋率是找到代碼庫未經(jīng)測試的部分的簡單方法。它并不會告訴你的測試好不好。
在 Python 中我們已經(jīng)有了一個不錯的覆蓋率工具來幫助我們。你可以在實驗樓環(huán)境中安裝它：
$ sudo pip3 install coverage
覆蓋率示例
$ coverage3 run mounttest.py
..
----------------------------------------------------------------------
Ran 2 tests in 0.013s
OK
$ coverage3 report -m
Name Stmts Miss Cover Missing
--------------------------------------------
mounttab2.py 22 7 68% 16, 25-30, 34
mounttest.py 14 0 100%
--------------------------------------------
TOTAL 36 7 81%
我們還可以使用下面的命令以 HTML 文件的形式輸出覆蓋率結(jié)果，然后在瀏覽器中查看它。
$ coverage3 html
知識點回顧：
本節(jié)了解了什么是單元測試，unittest 模塊怎么用，測試用例怎么寫。以及最后我們使用第三方模塊 coverage 進行了覆蓋率測試。
在實際生產(chǎn)環(huán)境中，測試環(huán)節(jié)是非常重要的的一環(huán)，即便志不在測試工程師，但以后的趨勢就是 DevOps，所以掌握良好的測試技能也是很有用的。
到此這篇關(guān)于淺談如何測試Python代碼的文章就介紹到這了,更多相關(guān)測試Python代碼內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
