Python自动化测试笔记（二）

Python自动化测试基础（二）

19.测试代码

编写函数或类时，还可为其编写测试。通过测试，可确定代码面对各种输入都能够按要求的那样工作。测试让你信心满满，深信即便有更多的人使用你的程序，它也 能正确地工作。在程序中添加新代码时，你也可以对其进行测试，确认它们不会破坏程序既有的行为。程序员都会犯错，因此每个程序员都必须经常测试其代码，在 用户发现问题前找出它们。

19.1 测试函数

要学习测试，得有要测试的代码。下面是一个简单的函数，它接受名和姓并返回整洁的姓名

def get_fromatted_name(first,last):
    full_name=first+' '+last
    return full_name.title()

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

def get_fromatted_name(first,last):
    full_name=first+' '+last
    return full_name.title()
print('输入q进行退出！')
while True:
    first=input('请输入你的姓：')
    if first=='q':
        break
    last=input('请输入你的名：')
    if last=='q':
        break
    from_name=get_fromatted_name(first,last)
    print('你的名字是：'+from_name)

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

19.1.1 单元测试和测试用例

Python标准库中的模块unittest 提供了代码测试工具。单元测试 用于核实函数的某个方面没有问题；测试用例 是一组单元测试，这些单元测试一起核实函数在各种情形下的 行为都符合要求。良好的测试用例考虑到了函数可能收到的各种输入，包含针对所有这些情形的测试。全覆盖式测试 用例包含一整套单元测试，涵盖了各种可能的函数使用方 式。对于大型项目，要实现全覆盖可能很难。通常，最初只要针对代码的重要行为编写测试即可，等项目被广泛使用时再考虑全覆盖。

19.1.2 可通过的测试

创建测试用例的语法需要一段时间才能习惯，但测试用例创建后，再添加针对函数的单元测试就很简单了。要为函数编写测试用例，可先导入模块unittest 以及要测试的函 数，再创建一个继承unittest.TestCase 的类，并编写一系列方法对函数行为的不同方面进行测试。

import unittest
from name_function import get_fromatted_name#导入该函数
class NameTestCase(unittest.TestCase):#创建一个类 必须继承unittest.TestCase类
    """测试get_fromatted_name方法"""
    def test_first_last_name(self):#测试类包含一个方法
        fromatted_name=get_fromatted_name('long','wenjie')#调用函数方法
        self.assertEqual(fromatted_name,'long wenjie')#调用assertEqual()断言方法，并向它传递formatted_name 和'Janis Joplin'
        """代码行self.assertEqual(formatted_name, 'Janis Joplin') 的意思是
说：“将formatted_name 的值同字符串'Janis Joplin' 进行比较，如果它们相等，就万事大吉，如果它们不相等，跟我说一声！”"""
unittest.main()

结果：

19.1.3 不能通过的测试

测试未通过时结果是什么样的呢？我们来修改get_formatted_name() ，使其能够处理中间名，但这样做时，故意让这个函数无法正确地处理像Janis Joplin这样只有名和姓的 姓名。

def get_fromatted_name(first,middle,last):
    full_name=first+' '+middle+' '+last
    return full_name

再次运行测试代码时会发现

其中包含的信息很多，因为测试未通过时，需要让你知道的事情可能有很多。第1行输出只有一个字母E （见❶），它指出测试用例中有一个单元测试导致了错误。接下来，我们 看到NamesTestCase 中的test_first_last_name() 导致了错误（见❷）。测试用例包含众多单元测试时，知道哪个测试未通过至关重要。在❸处，我们看到了一个标准 的traceback，它指出函数调用get_formatted_name('janis', 'joplin') 有问题，因为它缺少一个必不可少的位置实参。

19.1.4 测试未通过时怎么办

测试未通过时怎么办呢？如果你检查的条件没错，测试通过了意味着函数的行为是对的，而测试未通过意味着你编写的新代码有错。因此，测试未通过时，不要修改测试，而应 修复导致测试不能通过的代码：检查刚对函数所做的修改，找出导致函数行为不符合预期的修改。

19.1.5 添加新测试

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

import unittest
from name_function import get_fromatted_name#导入该函数
class NameTestCase(unittest.TestCase):#创建一个类 必须继承unittest.TestCase类
    """测试get_fromatted_name方法"""
    def test_first_last_name(self):#测试类包含一个方法
        fromatted_name=get_fromatted_name('long','wenjie')#调用函数方法
        self.assertEqual(fromatted_name,'long wenjie')#调用assertEqual()断言方法，并向它传递formatted_name 和'Janis Joplin'
        """代码行self.assertEqual(formatted_name, 'Janis Joplin') 的意思是
说：“将formatted_name 的值同字符串'Janis Joplin' 进行比较，如果它们相等，就万事大吉，如果它们不相等，跟我说一声！”"""
    def test_first_last_middle_name(self):
        fromatted_name=get_fromatted_name('long','wen','jie')
        self.assertEqual(fromatted_name,'long wen jie')
unittest.main()

我们将这个方法命名为test_first_last_middle_name() 。方法名必须以test_打头，这样它才会在我们运行test_name_function.py时自动运行。

19.2 测试类

在本章前半部分，你编写了针对单个函数的测试，下面来编写针对类的测试。很多程序中都会用到类，因此能够证明你的类能够正确地工作会大有裨益。如果针对类的测试通过 了，你就能确信对类所做的改进没有意外地破坏其原有的行为。

19.2.1 各种断言方法

Python在unittest.TestCase 类中提供了很多断言方法。前面说过，断言方法检查你认为应该满足的条件是否确实满足。如果该条件确实满足，你对程序行为的假设就得到了 确认，你就可以确信其中没有错误。如果你认为应该满足的条件实际上并不满足，Python将引发异常。 表11-1描述了6个常用的断言方法。使用这些方法可核实返回的值等于或不等于预期的值、返回的值为True 或False 、返回的值在列表中或不在列表中。你只能在继 承unittest.TestCase 的类中使用这些方法，下面来看看如何在测试类时使用其中的一个。

19.2.2 一个要测试的类

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

class AnonymousSurvey():
    """收集匿名调查问卷的答案"""
    def __init__(self, question):
        """存储一个问题，并为存储答案做准备"""
        self.question = question
        self.responses = []
    def show_question(self):
        """显示调查问卷"""
        print(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)

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

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

19.2.3 测试AnonymousSurvey 类

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)
unittest.main()

19.2.4 方法setUp()

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