编写函数或类时,还可为其编写测试。通过测试,可确定代码面对各种输入都能够按要求的那样工作。程序员都会犯错,因此每个程序员都必须经常测试其代码,在用户发现问题前找出它们。
测试函数
新建一个name_function.py
文件,进行加单的测试:
def get_name(first,last):
"""拼写出全名"""
full_name=first+' '+last
return full_name.title()
z在文件names.py
中运行name_function.py
from name_function import get_name
print("Enter 'q' at any time to quit.")
while True:
first = fbgbf
if first == 'q':
break
last=gbg
if last == 'q':
break
formatted_name=get_name(first,last)
print("\tNeatly formatted name: " + formatted_name + '.')
单元测试和测试用例
Python标准库中的模块unittest
提供了代码测试工具。单元测试
用于核实函数的某个方面没有问题;测试用例
是一组单元测试,这些单元测试一起核实函数在各种情形下的 行为都符合要求。良好的测试用例考虑到了函数可能收到的各种输入,包含针对所有这些情形的测试。全覆盖式测试
用例包含一整套单元测试,涵盖了各种可能的函数使用方 式。对于大型项目,要实现全覆盖可能很难。通常,最初只要针对代码的重要行为编写测试即可,等项目被广泛使用时再考虑全覆盖。
可通过的测试
创建测试用例的语法需要一段时间才能习惯,但测试用例创建后,再添加针对函数的单元测试就很简单了。要为函数编写测试用例,可先导入模块unittest
以及要测试的函数,再创建一个继承unittest.TestCase
的类,并编写一系列方法对函数行为的不同方面进行测试。
下面是一个只包含一个方法的测试用例,它检查函数get_formatted_name()
在给定名和姓时能否正确地工作:
import unittest
from name_function import get_name
class NamesTestCase(unittest.TestCase): #创建可包含get_nname()单元测试的类
"""测试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.000s #测试运行的时间
OK #表示测试通过
不能通过的测试
修改要测试的name_function.py
文件
def get_name(first,middle,last):
"""打印名字"""
full_name = first + ' ' + middle + ' ' + last
return full_name.title()
因为多了中间名字这个属性,就会显示错误信息:
E
======================================================================
ERROR: test_first_last_name (__main__.NamesTestCase)
能够正确地处理像Janis Joplin这样的姓名吗?
----------------------------------------------------------------------
Traceback (most recent call last):
File "/Users/jiemo/Documents/Python/day0/test_name.py", line 8, in test_first_last_name
formatted_name = get_name('janis', 'joplin')
TypeError: get_name() missing 1 required positional argument: 'last'
----------------------------------------------------------------------
Ran 1 test in 0.001s
测试类
下表描述了6个常用的断言方法。使用这些方法可核实返回的值等于或不等于预期的值、返回的值为True
或False
、返回的值在列表中或不在列表中。你只能在继承unittest.TestCase
的类中使用这些方法,下面来看看如何在测试类时使用其中的一个。
unittest Module中的断言方法
方法 | 用途 |
---|---|
assertEqual(a, b) | 核实a != b |
assertTrue(x)col 2 is | 核实x 为True |
assertFalse(x) | 核实x 为False |
assertIn(item , list ) | 核实 item 在 list 中 |
assertNotIn(item , list ) | 核实 item 不在 list 中 |
一个要测试的类
class AnonymousSurvey():
"""收集匿名调查问卷的答案"""
def __int__(self,question):
super(AnonymousSurvey, self).__init__()
self.question = question
self.responses=[]
def show_question(self):
"""显示调查问卷"""
print(question)
def store_response(self, new_response):
"""存储单份调查答卷"""
self.responses.append(new_response)
def show_results(self):
"""显示收集到的所有答卷"""
print("Survey results:")
for response in responses:
print('- ' + response)
from survey import AnonymousSurvey
#定义一个问题,并创建一个表示调查的AnonymousSurvey对象
question = "What language did you first learn to speak?"
my_survey = AnonymousSurvey(question)
#显示问题并存储答案
my_survey.show_question()
print("Enter 'q' at any time to quit.\n")
while True:
response = input("Language: ")
if response == 'q':
break
my_survey.store_response(response)
# 显示调查结果
print("\nThank you to everyone who participated in the survey!")
my_survey.show_results()
AnonymousSurvey
类可用于进行简单的匿名调查。
方法 setUp()
使用setUp()
来创建一个调查对象和一组答案,供方法test_store_single_response()
和test_store_three_responses()
使用:
import unittest
from survey import AnonymousSurvey
class TestAnonymousSurvey(unittest.TestCase):
"""针对AnonymousSurvey类的测试"""
def setUp(self):
"""创建一个调查对象和一组答案,供使用的测试方法使用"""
question = "What language did you first learn to speak?"
self.my_survey = AnonymousSurvey(question)
self.responses = ['English', 'Spanish', 'Mandarin']
def test_store_single_response(self):
"""测试单个答案会被妥善地存储"""
self.my_survey.store_response(self.responses[0])
self.assertIn(self.responses[0], self.my_survey.responses)
def test_store_three_responses(self):
"""测试三个答案会被妥善地存储"""
for response in self.responses:
self.my_survey.store_response(responses)
for response in self.responses:
self.assertIn(response, self.my_survey.responses)
unittest.main()
方法setUp()
做了两件事情:创建一个调查对象self.my_survey
;创建一个答案列表self.responses
;
测试自己编写的类时,方法setUp()
让测试方法编写起来更容易:可在setUp()
方法中创建一系列实例并设置它们的属性,再在测试方法中直接使用这些实例。相比于在每个 测试方法中都创建实例并设置其属性,这要容易得多。