Python單元測試簡單示例

更新時間：2018年07月03日 11:52:19 作者：ycwang522

這篇文章主要介紹了Python單元測試,結合實例形式分析了Python單元測試的簡單定義、使用方法及相關操作注意事項,需要的朋友可以參考下

本文實例講述了Python單元測試方法。分享給大家供大家參考，具體如下：

Eric書中《Python編程從入門到實踐》中的一個例子。

《Python編程從入門到實踐》隨書源碼可至此下載：http://m.fzitv.net/books/582019.html。

首先定義了一個測試函數(shù)：

namefunction.py

#-*- coding:cp936 -*-
def get_formmed_name(first, last):
 """該函數(shù)根據(jù)姓和名生成一個完整的姓名"""
 full_name = first + ' ' + last
 return full_name.title()

然后編寫了一個測試該函數(shù)的模塊：

names.py

#-*- coding:cp936 -*-
from name_function import get_formmed_name
"""該文件用來測試姓名生成函數(shù)"""
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中的函數(shù)可以實現(xiàn)其功能。

最后是單元測試和測試用例的編寫。

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的版本中直接調(diào)用書上的主函數(shù)unittest.main()時程序無法通過，而改用以下調(diào)用方式即可