编写函数和类时,还可以为其编写测试。通过测试,可确定代码面对各种输入都能够按要求那样工作。在程序添加新代码时,你也可以对其进行测试,确认它们不会破坏程序既有的行为。程序员都会犯错,因此每个程序员都必须经常测试其代码,在用户发现问题前找出它们。
name_function.py
def get_formatted_name(first, last):
"""Generate a neatly formatted full name."""
full_name = first + ' ' + last
return full_name.title()
names.py
from name_function import get_formatted_name
print("Enter 'q' at any time to quit.")
while True:
first = input("\nPlease give me a first name: ")
if first == 'q':
break
last = input("Please give me a lst name: ")
if last == 'q':
break
formatted_name = get_formatted_name(first, last)
print("\tNeatly formatted name: " + formatted_name + '.')
Python标准库中的模块unittest提供了代码测试工具。单元测试用于核实函数的某个方面没有问题;测试用例是一组单元测试,这些单元测试一起核实函数在各种情形下的行为都符合要求。良好的测试用例考虑到了函数可能的各种输入,包含针对所有这些情形的测试。全覆盖式测试用例包含一整套单元测试,涵盖了各种可能的函数使用方式。对于大型项目,要实现全覆盖可能很难。通常,最初只要针对代码的重要行为编写测试即可,等项目被广泛使用时再考虑全覆盖。
创建测试用例的语法需要一段时间才能习惯,但测试用例创建后,再针对函数的单元测试就很简单了。要为函数编写测试用例,可先导入模块unittest以及要测试的函数,再创建一个继承unittest.TestCase的类,并编写一系列方法对函数行为的不同方面进行测试。
下面时一个只包含一个方法的测试用例,它检查函数get_formatted_name( )在给定名和姓时能否正确地工作:
test_name_function.py
import unittest
from name_function import get_formatted_name
class NamesTestCase(unittest.TestCase):
"""测试name_function.py"""
def test_first_last_name(self):
"""能够正确地处理像Janis Joplin这样的姓名吗?"""
formatted_name = get_formatted_name('janis','joplin')
self.assertEqual(formatted_name,'Janis Joplin')
unittest.main()
----------------------------------------------------------------------
Ran 1 test in 0.007s
OK
首先,我们导入了模块unittest和要测试的函数get_formatted_name( )。我们创建了一个名为NamesTestCase的类,用于包含一系列针对get_formatted_name( )的单元测试。你可随便给这个类命名,但最好让它看起来与要测试的函数相关,并包含字样Test。这个类必须继承unittest.TestCase类,这样Python才知道如何运行你编写的代码。
NamesTestCase只包含一个方法,用于测试get_formatted_name( )的一个方面。我们将这个方法命名为test_first_last_name( ),因为我们要核实的时只有名和姓的姓名能够被正确地格式化。我们运行test_name_function.py时,所有以test_打头的方法都将自动运行。在这个方法中,我们调用了要测试的函数,并存储了要测试的返回值。在这个示例中,我们使用实参‘janis’和‘joplin’调用get_formatted_name( ),并将结果存储到变量formatted_name中。
我们使用了unittest类最有用的功能之一:一个断言方法。断言方法用来核实得到的结果是否与期望的一致。代码行self.assertEqual(formatted_name,’Janis Joplin’)的意思是说:“将formatted_name的值同字符串‘Janis Joplin’进行比较,如果它们相等,就万事大吉。
Unittest Module中断言的方法
方法 | Are |
---|---|
assertEqual(a,b) | 核实a==b |
assertNotEqual(a,b) | 核实a!=b |
assertTrue(x) | 核实X为true |
assertFalse(x) | 核实X为false |
assertIn(item,list) | 核实item在list中 |
assertNotIn(item,list) | 核实item不在list中 |
类的测试与函数的测试相似——你所做的大部分动作都是测试类中方法的行为,但存在一些不同之处。
class AnonymousSurvey():
"""手机匿名调查问卷的答案"""
def __init__(self,question):
"""存储一个问题,并为村存储答案做准备"""
self.question = quesiton
self.responses = []
def show_question(self):
"""显示调查问卷"""
print(question)
def show_response(self,new_response):
"""存储单份调查答卷"""
self.response.append(new_response)
def show_results(self):
"""显示收集到的所有答案"""
print("Survey results:")
for response in responses:
print('- ' + response)