python之unitest单元测试
Eric书中《Python编程从入门到实践中的一个例子》。
首先定义了一个测试函数:
namefunction.py #-*- coding:cp936 -*- def get_formmed_name(first, last): """该函数根据姓和名生成一个完整的姓名""" full_name = first + ' ' + last return full_name.title()
然后编写了一个测试该函数的模块:
#-*- 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.main()
通过对以上单元测试模块分析:
导入单元测试类unittest
导入要测试的函数,本例为name_function模块中的get_formatted_name() 函数
创建一个继承于unittest.TestCase的类
在类中定义一系列方法对函数的行为进行不同方面的测试,需要注意的是一个测试用例应该只测试一个方面,测试目的和测试内容应很明确。主要是调用assertEqual、assertRaises等断言方法判断程序执行结果和预期值是否相符。
拼搏在路上