编写函数或类时,还可为其编写测试。通过测试,可确定代码面对各种输入都能够按要求的那样工作。
1.1测试函数
1.1.1 可通过的测试
要为函数编写测试用例,可先导入模块
unittest
以及要测试的函 数,再创建一个继承unittest.TestCase
的类,并编写一系列方法对函数行为的不同方面进行测试。
下面是一个只包含一个方法的测试用例,它检查函数
get_formatted_name()
在给定名和姓时能否正确地工作:
首先是需要被测试的函数:
def get_formatted_name(first, last, middle=''):
"""生成整洁的姓名"""
if middle:
full_name = first + ' ' + middle + ' ' + last
else:
full_name = first + ' ' + last
return full_name.title()
然后是测试代码:
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')
if __name__=='__main__':
unittest.main()
结果如下:
所有以test开头的方法都将自动运行。
self.assertEqual(formatted_name, 'Janis Joplin')是unittest 类最有用的功能之一——断言:用来核实得到的结果是否与期望的结果一致。
1.1.2 不能通过的测试
如果不一致会是什么结果呢?我们可以将测试函数修改:
formatted_name = get_formatted_name('janis', 'joplin','a')
结果显示:
1.2 测试类
前面我们学习了编写了针对单个函数的测试,下面来编写针对类的测试。
1.2.1 各种断言方法
断言方法检查你认为应该满足的条件是否确实满足。如果该条件确实满足,你对程序行为的假设就得到了确认,你就可以确信其中没有错误。如果你认为应该满足的条件实际上并不满足,Python
将引发异常。
![](https://i-blog.csdnimg.cn/blog_migrate/257fa9e972db98b8fb99d7ff3452640c.png)
1.2.2 一个要测试的类
下面来编写一个类进行测试。来看一个帮助管理收集一个问题的答案的类:
class AnonymousSurvey():
"""收集匿名调查问卷的答案"""
def __init__(self, question):
"""存储一个问题,并为存储答案做准备"""
self.question = question
self.responses = []
def show_question(self):
"""显示调查问卷"""
print(self.question)
def store_response(self, new_response):
"""存储单份调查答卷"""
self.responses.append(new_response)
def show_results(self):
"""显示收集到的所有答卷"""
print("Survey results:")
for response in self.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()
下面我们来编写一个测试,如果用户面对调查问题时只提供了一个答案,这个答案也能被妥善地存储。
import unittest
from survey import AnonymousSurvey
class TestAnonmyousSurvey(unittest.TestCase):
"""针对AnonymousSurvey类的测试"""
def test_store_single_response(self):
"""测试单个答案会被妥善地存储"""
question = "What language did you first learn to speak?"
my_survey = AnonymousSurvey(question)
my_survey.store_response('English')
self.assertIn('English', my_survey.responses)
if __name__=='__main__':
unittest.main()
1.2.3 方法SetUp()
在前面的
test_survey.py
中,我们在每个测试方法中都创建了一个
AnonymousSurvey
实例,并在每个方法中都创建了答案。
unittest.TestCase
类包含方法
setUp()
,让我 们只需创建这些对象一次,并在每个测试方法中使用它们。如果你在TestCase
类中包含了方法
setUp()
,
Python
将先运行它,再运行各个以
test_
打头的方法。这样,在你编写 的每个测试方法中都可使用在方法setUp()
中创建的对象了。
下面使用
setUp()
来创建一个调查对象和一组答案,供方法
test_store_single_response()
和
test_store_three_responses()
使用:
import unittest
from survey import AnonymousSurvey
class TestAnonymousSurvey(unittest.TestCase):
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(response)
for response in self.responses:
self.assertIn(response, self.my_survey.responses)
if __name__ == '__main__':
unittest.main()
方法
setUp()
做了两件事情:创建一个调查对象
;创建一个答案列表
。存储这两样东西的变量名包含前缀
self
(即存储在属性中),因此可在这个类的任何 地方使用。这让两个测试方法都更简单,因为它们都不用创建调查对象和答案。
方法test_store_three_response()
核实
self.responses
中的第一个答案 ——self.responses[0]
——
被妥善地存储,而方法
test_store_three_response()
核实
self.responses
中的全部三个答案都被妥善地存储。
测试自己编写的类时,方法
setUp()
让测试方法编写起来更容易:可在
setUp()
方法中创建一系列实例并设置它们的属性,再在测试方法中直接使用这些实例。相比于在每个测试方法中都创建实例并设置其属性,这要容易得多。