python测试代码怎么写_python基础教程之第一章简单测试代码|python基础教程|python入门|python教程...-CSDN博客

https://www.xin3721.com/eschool/pythonxin3721/

1.1 测试函数

下面是一个简单的函数，它接受名和姓，并返回整洁的姓名：

name_function.py

defget_formatted_name(first,last):"""Generate a neatly formatted full name"""full_name= first + " " +lastreturn full_name.title()

函数get_formatted_name()将名和姓合并成姓名，在姓和名之间加上一个空格，并将它们的首字母大写，再返回结果。为核实get_formatted_name()像期望中的那样工作，我们比编写一个使用这个函数的程序。程序names.py让用户输入名和姓，并显示整洁的全名。

name.py

form name_function importget_formatted_nameprint("Enter"q"at any time to quit.")whileTrue:

first= input("/nPlease give me a first name:")if first == 'q':breaklast= input("Please give me a last name:")if last == 'q':breakformatted_name=get_formatted_name(first,last)print("/tNeatly formatted name:" + formatted_name + ".")

这个程序从name_function.py中导入get_formatted_name（）。用户可输入一系列的名和姓，并看到格式整洁的全名：

Enter 'q'at any time to quit.

Please give me a first name: janis

Please give ne a last name: joplin

Neatly formatted name: Janis Joplin.

Please give me a first name: bob

Please give ne a last name: dylan

Neatly formatted name: Bob Dylan.

Please give me a first name: q

从上述输出可知，合并得到的姓名正确无误。现在假设我们要修改get_formatted_name(),使其还能够处理中间名。这样做时，我们要确保不破坏这个函数处理只有名和姓的姓名的方式。为此，我们可以在每次修改get_formatted_name()都进行测试：运行程序names.py，并输入像Janis Joplin这样的姓名，但这太繁琐了。所幸Python提供一种自动测试函数输出的高效方式。倘若我们对get_formatted_name()进行自动测试，就能始终信心满满，确信这个函数提供给我们测试过的姓名时，它都能正确地工作。

1.2测试用例

下面是一个只包含一个方法的测试用例，它检查函数get_formatted_name()在给定名和姓时能否正确地工作：

test_name_function.py

1 importunittest2 from name_function importget_formatted_name3

4 classNamesTestCase(unittest.TestCase):5 """测试name_function.py"""

6 deftest_first_last_naem(self):7 """能够正确地处理像 Janis Joplin 这样的姓名吗？"""

8 formatted_name = get_formattde_name("janis", "joplin")9 self.assertEqual(formatted_name, "Janis Joplin")10

11 unittest.main()

1.3 不能通过的测试

上面的代码只能判断只有姓和名时的一方面，对于有中间名的就不能够通过测试了。比如我们来修改get_formatted_name()，使其能够处理中间名。

下面是函数get_formatted_name()的新版本，它要求通过一个实参指定中间名：

name_function.py

1 defget_formatted_name(first, middle,last):2 """生成整洁的姓名"""

3 full_name = first + " " + middle + " " +last4 return full_name.title()

修改后再测试，我们发现它不再能正确地处理只有名和姓的姓名，因为它缺少了一个实参，那怎么解决这个问题呢？其实只需要把中间名变成可选的就行了，要将中间名变成可选的，可在函数定义中将形参middle移到形参列表末尾，并将其默认值指定为一个空字符串。我们还要添加一个if测试，以便根据是否提供了中间名相应地创建姓名：

name_function.py

def get_formatted_name(fjirst,last,middle=""):"""生成整洁的姓名"""

ifmiddle:

full_name= first + ' ' + middle + ' ' +lastelse:

full_name= first + ' ' +lastreturn full_name.title()

再次运行test_name_function.py 测试用例就能通过了。

1.4 添加新的测试

确定get_formatted_name() 又能正确地处理简单的名和姓后，我们再编一个测试，用于测试包含中间名的姓名。为此，我们在NameTestCase类中再添加一个方法：

mport unittestfrom name_function importget_formatted_nameclassNamesTestCase(unittest.TestCase):"""测试name_function.py"""

deftest_first_last_naem(self):"""能处理像 Janis Joplin 这样的姓名吗？"""formatted_name= get_formattde_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()

方法名必须以test_打头，这样它才会在我们运行test_name_function.py时自动运行。

1.5Python中unittest Moudle中的断言方法

上表1.5.1描述了6个常用的断言方法。使用这些方法可核实返回的值等于或者不等于预期值、返回的值为True或False、返回的值在列表中或不在列表中。你只能在继承unittest.TestCase的类中使用这些方法。

1.6 一个要测试的类

类的测试与函数的测试相似----你所做的大部分工作都是测试类中方法的行为，但存在一些不同之处，下面来编写一个类进行测试。来看一个帮助管理匿名调查的类：

survey.py

classAnonymousSurvey():""""收集匿名调查问卷的答案"""

def __init__(self,question):"""存储一个问题，并为存储答案做准备"""self.question=question

self.responses=[]defshou_question(self):"""显示调查问卷"""

print(self.question)defstore_response(self,new_response):"""存储单份调查答卷"""self.responses.append(new_response)defshow_results(self):"""显示收集到的所有答卷"""

print("Survey results:")for response inresponses:print('-' + response)

为证明AnonymousSurvey类能正常地工作，我们来编写一个使用它的程序：

language_survey.py

from survey importAnonymousSurvey#定义一个问题，并创建一个表示调查的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")whileTrue:

response= input("Language:")if response == 'q'

breakmy_survey.store_response(response)#显示调查结果

print("\nThank you to everyone who participated in the survey!")

my_survey.show_results()

1.61.测试AnonymousSurvey类

下面来编写一个测试，对AnonymousSurvey类的行为的一个方面进行验证：如果用户面对调查问题时只提供一个答案，这个答案也能被妥善地储存。为此，我们将在这个答案被存储后，使用方法assertIn()来核实它包含在答案列表中：

test_survey.py

importunittestfrom survey importAnonymousSurveyclassTestAnonymousSurvey(unittest.TestCase):"""针对AnonymousSurvey类的测试"""

deftest_store_sigle_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()

这很好，但只能收集一个答案的调查用途不大。下面来核实用户提供三个答案时，它们也将被妥善地存储。为此，我们在TestAnonymousSurvey中再添加一个方法：

importunittestfrom survey importAnonymousSurveyclassTestAnonymousSurvey(unittest.TestCase):"""针对AnonymousSurvey类的测试"""

deftest_store_sigle_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)deftest_store_three_responses(self):"""测试三个答案会被妥善地存储"""question- "What language did you first lenar to spesk?"my_survey=AnonymousSurvey(question)

responses= ['English','Spanish','Mandarin']for response inresponses:

my_survey.store_response(response)for response inresponses:

self.assertIn(response, my_survey.response)

unittest.main()

我们定义了一个包含三个不同答案的列表，再对其中每个答案都调用store_response()。存储这些答案后，我们使用一个循环来确认每个答案都包含在my_survey.response中。

前述做法的效果很好，但这些测试有些重复的地方。下面我们用unittest的另一项功能来提高它们的效率

importunittestfrom survey importAnonymousSurveyclassTestAnonymousSurvey(unittest.TestCase):"""针对AnonymousSurvey的测试"""

defsetup(self):"""创建一个调查对象和一组答案，供使用的测试方法使用"""question= "What language did you first learn to speak?"self.my_survey=AnonymousSurvey(question)

self.responses= ["English", "Spanish", "Mandarin"]deftest_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 inself.responses: