前言:
编写函数或者类时,需要编写测试代码,来保证其的功能运行是否按预期的那样工作。在程序添加新的代码时,用来测试是否会破坏本身的功能。
我们使用python自带的unittest模块来测试代码。
编写函数:
def get_formatted_name(first,last,middle=''):"""生成整洁的姓名"""
ifmiddle:
full_name= first + ' '+ middle +' '+lastelse:
full_name= first + ' ' +lastreturn full_name.title()
编写测试:
importunittestfrom name_function importget_formatted_nameclassNameTestCase(unittest.TestCase):"""测试name_function.py"""
deftest_first_last_name(self):"""能够正确地处理像Janis Joplin这样的姓名吗?"""formatted_name= get_formatted_name('janis','joplin')
self.assertEqual(formatted_name,'Janis Joplin')deftest_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()
编写类:
classAnonymousSurvey():"""收集匿名调查问卷的答案"""
def __init__(self, question):"""存储一个问题,并为存储答案做准备"""self.question=question
self.responses=[]defshow_question(self):"""显示调查问卷"""
print(self.question)defstore_response(self, new_response):"""存储单份调查答卷"""self.responses.append(new_response)defshow_results(self):"""显示收集到的所有答卷"""
print("Survey results")for response inself.responses:print('-'+ response)
编写测试:
importunittestfrom survey importAnonymousSurveyclassTestAnonymousSurvey(unittest.TestCase):"""针对AnonymousSurvey类的测试"""
deftest_store_single_response(self):"""测试三个答案会被妥善地存储"""question= "What language did you first learn to speak?"my_survey=AnonymousSurvey(question)
lists= ['English','China','Japanese']for s inlists:
my_survey.store_response(s)
self.assertIn(s, my_survey.responses)
unittest.main()
总结:
unittest模块不仅支持单元测试,在接口和功能测试也能直接套用其中的方法。编写unittest必须添加继承unittest.TestCase类,测试的函数必须以test*开头,以及使用unittest.mian方法来运行此测试模块。