《python编程：从入门到实践》第十一章 测试代码

编写函数或类时，还可为其编写测试。通过测试，可确定代码面对各种输入都能够按要求的那样工作。

目录

1.1测试函数

1.1.1 可通过的测试

1.1.2 不能通过的测试

1.2 测试类

1.2.1 各种断言方法

 1.2.2 一个要测试的类

1.2.3 方法SetUp()

---

1.1测试函数

1.1.1 可通过的测试

要为函数编写测试用例，可先导入模块 unittest 以及要测试的函 数，再创建一个继承unittest.TestCase 的类，并编写一系列方法对函数行为的不同方面进行测试。

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

首先是需要被测试的函数：

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

然后是测试代码：

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

if __name__=='__main__':
 unittest.main()

结果如下：

 所有以test开头的方法都将自动运行。

self.assertEqual(formatted_name, 'Janis Joplin')是unittest 类最有用的功能之一——断言：用来核实得到的结果是否与期望的结果一致。

1.1.2 不能通过的测试

 如果不一致会是什么结果呢？我们可以将测试函数修改：

   formatted_name = get_formatted_name('janis', 'joplin'，'a')

结果显示：

1.2 测试类

前面我们学习了编写了针对单个函数的测试，下面来编写针对类的测试。

1.2.1 各种断言方法

断言方法检查你认为应该满足的条件是否确实满足。如果该条件确实满足，你对程序行为的假设就得到了确认，你就可以确信其中没有错误。如果你认为应该满足的条件实际上并不满足，Python 将引发异常。

 1.2.2 一个要测试的类

下面来编写一个类进行测试。来看一个帮助管理收集一个问题的答案的类：

class AnonymousSurvey():
    """收集匿名调查问卷的答案"""
    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 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()

下面我们来编写一个测试，如果用户面对调查问题时只提供了一个答案，这个答案也能被妥善地存储。

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)
if __name__=='__main__':
    unittest.main()

1.2.3 方法SetUp()

在前面的 test_survey.py 中，我们在每个测试方法中都创建了一个 AnonymousSurvey 实例，并在每个方法中都创建了答案。 unittest.TestCase 类包含方法 setUp() ，让我 们只需创建这些对象一次，并在每个测试方法中使用它们。如果你在TestCase 类中包含了方法 setUp() ， Python 将先运行它，再运行各个以 test_ 打头的方法。这样，在你编写 的每个测试方法中都可使用在方法setUp() 中创建的对象了。

下面使用 setUp() 来创建一个调查对象和一组答案，供方法 test_store_single_response() 和 test_store_three_responses() 使用：

import unittest
from survey import AnonymousSurvey

class TestAnonymousSurvey(unittest.TestCase):

 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)

if __name__ == '__main__':
 unittest.main()

方法 setUp() 做了两件事情：创建一个调查对象 ；创建一个答案列表 。存储这两样东西的变量名包含前缀 self （即存储在属性中），因此可在这个类的任何 地方使用。这让两个测试方法都更简单，因为它们都不用创建调查对象和答案。

方法test_store_three_response() 核实 self.responses 中的第一个答案 ——self.responses[0] —— 被妥善地存储，而方法 test_store_three_response() 核实 self.responses 中的全部三个答案都被妥善地存储。

测试自己编写的类时，方法 setUp() 让测试方法编写起来更容易：可在 setUp() 方法中创建一系列实例并设置它们的属性，再在测试方法中直接使用这些实例。相比于在每个测试方法中都创建实例并设置其属性，这要容易得多。