第一章 简单测试代码

1.1  测试函数

 

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

　　name_function.py

def get_fORMatted_name(first,last):
    """Generate a neatly formatted full name"""
    full_name = first + " " +last
    return full_name.title()

View Code

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

　　name.py

form name_function import get_formatted_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("Please give me a last name:")
    if last == 'q':
        break

    formatted_name = get_formatted_name(first,last)
    print("/tNeatly formatted name: " + formatted_name + ".")

View Code

　　这个程序从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

View Code

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

1.2测试用例

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

test_name_function.py

 1 import unittest
 2 from name_function import get_formatted_name
 3 
 4 class NamesTestCase(unittest.TestCase):
 5     """测试name_function.py"""
 6     def test_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()

View Code

1.3 不能通过的测试

上面的代码只能判断只有姓和名时的一方面，对于有中间名的就不能够通过测试了。比如我们来修改get_formatted_name()，使其能够处理中间名。
下面是函数get_formatted_name()的新版本，它要求通过一个实参指定中间名：

name_function.py

1 def get_formatted_name(first, middle,last):
2     """生成整洁的姓名"""
3     full_name = first + " " + middle + " " + last
4     return full_name.title()

View Code

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

name_function.py

def get_formatted_name(fjirst,last,middle=""):
    """生成整洁的姓名"""
    if middle:
        full_name = first + ' ' + middle + ' ' + last
    else:
        full_name = first + ' ' + last
    return full_name.title()

View Code

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

1.4 添加新的测试

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

mport unittest
from name_function import get_formatted_name

class NamesTestCase(unittest.TestCase):
    """测试name_function.py"""
    def test_first_last_naem(self):
    """能处理像 Janis Joplin 这样的姓名吗？"""
        formatted_name = get_formattde_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()    

View Code

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

1.5Python中unittest Moudle中的断言方法

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

1.6 一个要测试的类

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

survey.py

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

   def __init__(self,question):
    """存储一个问题，并为存储答案做准备"""
    self.question = question
    self.responses = []
   def shou_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 responses:
        print('-' + response)

View Code

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

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

View Code

 1.61.测试AnonymousSurvey类　　

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

test_survey.py

import unittest
from survey import AnonymousSurvey

class TestAnonymousSurvey(unittest.TestCase):
   """针对AnonymousSurvey类的测试"""
   def test_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()

View Code

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

import unittest
from survey import AnonymousSurvey

class TestAnonymousSurvey(unittest.TestCase):
   """针对AnonymousSurvey类的测试"""
   def test_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)
    
    def test_store_three_responses(self):
        """测试三个答案会被妥善地存储"""
        question - "What language did you first lenar to spesk?"
        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.response)

unittest.main()

View Code

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

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

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

View Code