Eric書中《Python編程從入門到實踐中的一個例子》。
首先定義了一個測試函數:
- namefunction.py
#-*- coding:cp936 -*-
def get_formmed_name(first, last):
"""該函數根據姓和名生成一個完整的姓名"""
full_name = first + ' ' + last
return full_name.title()
然後編寫了一個測試該函數的模塊:
- names.py
#-*- coding:cp936 -*-
from name_function import get_formmed_name
"""該文件用來測試姓名生成函數"""
print "Enter 'q' to quit any time."
while True:
first = raw_input("\nEnter first name:")
if first == 'q':
break
last = raw_input("Enter last name:")
if last == 'q':
break
formatted_name = get_formmed_name(first,last)
print "Formmated name:",formatted_name
通過測試,name_function中的函數可以實現其功能。
最後是單元測試和測試用例的編寫。
- test_name_function.py
# coding:utf-8
import unittest
from name_function import get_formmed_name
class NamesTestCase(unittest.TestCase):
"""測試name_function.py"""
def test_first_last_name(self):
"""能夠正確處理像Janis Joplin這樣的姓名嗎?"""
formatted_name = get_formmed_name('janis','joplin')
self.assertEqual(formatted_name,'Janis Joplin')
#注意這個地方,我在Python 2.7的版本中直接調用書上的主函數unittest.main()時程序無法通過,而改用以下調用方式即可
if __name__ == '__main__':
unittest
通過對以上單元測試模塊分析:
- 導入單元測試類unittest
- 導入要測試的函數,本例爲name_function模塊中的get_formatted_name() 函數
- 創建一個繼承於unittest.TestCase的類
- 在類中定義一系列方法對函數的行爲進行不同方面的測試,需要注意的是一個測試用例應該只測試一個方面,測試目的和測試內容應很明確。主要是調用assertEqual、assertRaises等斷言方法判斷程序執行結果和預期值是否相符。