10.1 测试函数
10.1.1 单元测试和测试用例
单元测试用于核实函数的某个方面没有问题；测试用例是一组单元测试，这些单元测试一起核实函数在各种情形下的行为都符合要求。
10.1.2 可通过的测试
要为函数编写测试用例，可先导入模块unittest以及要测试的函数，再创建一个继承unittest.TestCase的类，并编写一系列方法对函数行为的不同方面进行测试。
example：
import unittest
from name_function import get_formatted_name
class NamesTestCase(unittest.TestCase): #编写类继承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() #代码行unittest.main()让Python运行这个文件中的测试
我们运行test_name_function.py时，所有以test_打头的方法都将自动运行
代码行self.assertEqual(formatted_name, 'Janis Joplin')的意思是说：
“将formatted_name的值同字符串'Janis Joplin'进行比较，如果它们相等，就万事大吉，如果它
们不相等，跟我说一声！”
10.1.3 不能通过的测试
代码行unittest.main()让Python运行这个文件中的测试
def get_formatted_name(first, middle, last):
"""生成整洁的姓名"""
full_name = first + ' ' + middle + ' ' + last
return full_name.title()
#结果：
E
======================================================================
ERROR: test_first_last_name (__main__.NamesTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "test_name_function.py", line 8, in test_first_last_name
formatted_name = get_formatted_name('janis', 'joplin')
TypeError: get_formatted_name() missing 1 required positional argument: 'last'
----------------------------------------------------------------------
Ran 1 test in 0.000s
FAILED (errors=1)
第1行输出只有一个字母E（见Ø），它指出测试用例中有一个单元测试导致了错误。接下来，我们看到
NamesTestCase中的test_first_last_name()导致了错误（见）。测试用例包含众多单元测试时，
知道哪个测试未通过至关重要。在处，我们看到了一个标准的traceback，它指出函数调用
get_formatted_name('janis', 'joplin')有问题，因为它缺少一个必不可少的位置实参。
我们还看到运行了一个单元测试（见）。最后，还看到了一条消息，它指出整个测试用例
都未通过，因为运行该测试用例时发生了一个错误（见）。这条消息位于输出末尾，让你一眼
就能看到——你可不希望为获悉有多少测试未通过而翻阅长长的输出。
10.1.4 测试未通过时怎么办
根据测试结果修改原有的函数
example：10.1.3中将中间名设置为可选的
def get_formatted_name(first, last, middle=''):
"""生成整洁的姓名"""
if middle:
full_name = first + ' ' + middle + ' ' + last
else:
full_name = first + ' ' + last
return full_name.title()
#结果：
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
10.1.5 添加新的测试
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')
def test_first_last_middle_name(self):
"""能够正确地处理像Wolfgang Amadeus Mozart这样的姓名吗？"""
formatted_name = get_formatted_name(
'wolfgang', 'mozart', 'amadeus')
self.assertEqual(formatted_name, 'Wolfgang Amadeus Mozart')
unittest.main()
们将这个方法命名为test_first_last_middle_name()。方法名必须以test_打头，这样它才
会在我们运行test_name_function.py时自动运行。
10.2 测试类
10.2.1 6中常用的断言方法
| 方法 | 用途 |
|---|---|
| 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中 |
10.2.2 一个要测试的类
survey.py
class AnonymousSurvey():
"""收集匿名调查问卷的答案"""
def __init__(self, question):
"""存储一个问题，并为存储答案做准备"""
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)
language_survey.py
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()
输出结果：
What language did you first learn to speak?
Enter 'q' at any time to quit.
Language: English
Language: Spanish
Language: English
Language: Mandarin
Language: q
Thank you to everyone who participated in the survey!
Survey results:
- English
- Spanish
- English
- Mandarin
10.2.3 测试AnonymousSurvery
test_survey.py
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)
unittest.main()
要测试类的行为，需要创建其实例
结果：
.
----------------------------------------------------------------------
Ran 1 test in 0.001s
OK
核实用户提供三个答案时，它们也将被妥善地存储。为此，我们在TestAnonymousSurvey中再添加一个方法：
import unittest
from survey import AnonymousSurvey
class TestAnonymousSurvey(unittest.TestCase):
"""针对AnonymousSurvey类的测试"""
def test_store_single_response(self):
"""测试单个答案会被妥善地存储"""
--snip--
def test_store_three_responses(self):
"""测试三个答案会被妥善地存储"""
question = "What language did you first learn to speak?"
my_survey = AnonymousSurvey(question)
responses = ['English', 'Spanish', 'Mandarin']
for response in responses:
my_survey.store_response(response)
for response in responses:
self.assertIn(response, my_survey.responses)
unittest.main()
10.2.4 方法setUp（）
如果你在TestCase类中包含了方法setUp()，Python将先运行 它，再运行各个以test_打头的方法。这样，在你编写的每个测试方法中都可使用在方法setUp() 中创建的对象了。
测试自己编写的类时，方法setUp()让测试方法编写起来更容易：可在setUp()方法中创建一
系列实例并设置它们的属性，再在测试方法中直接使用这些实例。相比于在每个测试方法中都创
建实例并设置其属性，这要容易得多。
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(response)
for response in self.responses:
self.assertIn(response, self.my_survey.responses)
unittest.main()
注意： 运行测试用例时，每完成一个单元测试，Python都打印一个字符：测试通过时打印一个
句点；测试引发错误时打印一个E；测试导致断言失败时打印一个F。