def get_name(first,last):
full_name = first + ' '+last
return full_name.title()
from name import get_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('\nplease give me a last name:')
if last == 'q':
break
formatted_name = get_name(first,last)
print('\tNeatly formatted name:'+ formatted_name )
Python标准库中的模块unittest提供了代码测试工具。
单元测试用于核实函数的某个方面没有问题；测试用例是一组单元测试，这些单元测试一起核实函数在各种情形下的行为都符合要求。良好的测试用例考虑到了函数可能收到的各种输入，包含针对所有这些情形的测试。
全覆盖式测试用例包含一整套单元测试，涵盖了各种可能的函数使用方式。对于大型项目，要实现全覆盖可能很难。通常，最初只要针对代码的重要行为编写测试即可，等项目被广泛使用时再考虑全覆盖。
要为函数编写测试用例，可先导入模块unittest以及要测试的函数，再创建一个继承unittest.TestCase的类，并编写一系列方法对函数行为的不同方面进行测试。
import unittest
from name import get_name
class NameTestCase(unittest.TestCase):
'''测试name_function.py'''
def test_first_last_name(self):
#方法名必须以test_打头，这样它才会在我们运行py时自动运行
'''能够正确地处理像Janis Joplin这样的姓名吗'''
formatted_name= get_name('Janis','Joplin')
#调用了要测试的函数，并存储了要测试的返回值
self.assertEqual(formatted_name,'Janis Joplin')
#断言方法用来核实得到的结果是否与期望的结果一致。
'''运行这个文件中的测试'''
unittest.main()
输出：
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
第1行的句点表明有一个测试通过了。接下来的一行指出Python运行了一个测试，消耗的时
间不到0.001秒。最后的OK表明该测试用例中的所有单元测试都通过了。
测试未通过时结果是什么样的呢？我们来修改get_name()，使其能够处理中间名，但这样做时，故意让这个函数无法正确地处理像Janis Joplin这样只有名和姓的姓名。
def get_name(first,middle,last):
full_name = first + ' '+ middle+' '+last
return full_name.title()
输出：
E
======================================================================
ERROR: test_first_last_name (__main__.NameTestCase)
能够正确地处理像Janis Joplin这样的姓名吗
----------------------------------------------------------------------
Traceback (most recent call last):
File "f:\python练习\python\A基础语法\10_测试代码.py", line 22, 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 #单元测试
FAILED (errors=1)
第1行输出只有一个字母E，它指出测试用例中有一个单元测试导致了错误。接下来，我们看到
NamesTestCase中的test_first_last_name()导致了错误。测试用例包含众多单元测试时，知道哪个测试未通过至关重要。我们看到了一个标准的traceback，它指出函数调用get_formatted_name(‘janis’, ‘joplin’)有问题，因为它缺少一个必不可少的位置实参。我们还看到运行了一个单元测试。最后，还看到了一条消息，它指出整个测试用例都未通过，因为运行该测试用例时发生了一个错误。
def get_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
import unittest
from name import get_name
class NameTestCase(unittest.TestCase):
'''测试name_function.py'''
def test_first_last_name(self):
'''能够正确地处理像Janis Joplin这样的姓名吗'''
formatted_name= get_name('Janis','Joplin')
#调用了要测试的函数，并存储了要测试的返回值
self.assertEqual(formatted_name,'Janis Joplin')
#断言方法用来核实得到的结果是否与期望的结果一致。
**def test_first_last_middle(self): #新测试
'''能够正确地处理像Uzi Theshy Deft这样的姓名吗？'''
formatted_name = get_name('uzi','deft',middle='theshy')
self.assertEqual(formatted_name,'Uzi Theshy Deft')**
'''运行这个文件中的测试'''
unittest.main()
输出：
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
unittest Module中的断言方法
class Survey(): #survey.py
'''收集匿名调查问卷的答案'''
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 Survey
#定义一个问题，并创建一个表示调查的AnonymousSurvey对象
question = "What language did you first learn to speak?"
my_survey = Survey(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 Survey
class TestSurvey(unittest.TestCase):
'''针对Survey类的测试'''
def test_store_single_response(self):
'''测试单个答案会被妥善存储'''
question = 'What language did you first learn to speak?'
my_survey = Survey(question)
my_survey.store_response('English')
self.assertIn('English',my_survey.responses)
def test_store_three_response(self):
'''测试三个答案会不会被妥善存储'''
question = 'What language did you first learn to speak?'
my_survey = Survey(question)
responses = ['China','English','Spanish']
for response in responses:
my_survey.store_response(response)
for response in responses:
self.assertIn(response,my_survey.responses)
unittest.main()
import unittest
from survey import Survey
class TestSurvey(unittest.TestCase):
'''针对Survey类的测试'''
def setUp(self):
'''创建一个调查对象和一组答案，供使用的测试方法使用'''
question = 'What language did you first learn to speak?'
self.my_survey = Survey(question)
self.responses = ['China','English','Spanish']
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_response(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()
方法setUp()做了两件事情：创建一个调查对象；创建一个答案列表。存储这两样东西的变量名包含前缀self（即存储在属性中），因此可在这个类的任何地方使用。这让两个测试方法都更简单，因为它们都不用创建调查对象和答案。
注意：
运行测试用例时，每完成一个单元测试，Python都打印一个字符：测试通过时打印一个句点；测试引发错误时打印一个E；测试导致断言失败时打印一个F。这就是你运行测试用例时，在输出的第一行中看到的句点和字符数量各不相同的原因。如果测试用例包含很多单元测试，需要运行很长时间，就可通过观察这些结果来获悉有多少个测试通过了。