Python3：测试代码

测试函数

单元测试和测试用例

• Python标准库中的模块unittest提供了代码测试工具
• 单元测试用于核实函数的某个方面没有问题，测试用例是一组单元测试；
• 全覆盖式测试用例包含一整套单元测试

测试函数

• 首先需要准备要测试的函数

# name_func.py
def get_formatted_name(first, last):
    full_name = first + ' ' + last
    return full_name.title()

• 编写测试用例

#test_name_func.py
import unittest
from name_func import get_formatted_name

class NamesTestCase(unittest.TestCase):
    """测试name_func.py"""
    def test_first_last_name(self):
        """能够正确地处理像Janis Joplin这样的姓名吗？ """
        formatted_name = get_formatted_name('janis', 'joplin')
        self.assertEqual(formatted_name, 'Janis Joplin')

unittest.main()

• 首先，导入模块unittest和要测试的函数get_formatted_name()
• 然后创建一个继承unittest.TestCase的类NamesTestCase，用于包含一系列针对测试函数的单元测试
• NamesTestCase包含一个方法test_first_last_name()，当我们运行test_name_func.py时，所有以test_开头的方法都将自动运行。
• 接下来调用unittest的断言方法assertEqual()，用以比较函数结果和正确结果，如果相等，则万事大吉，如果不相等就和我说一声。
• 测试成功，将输出如下

.
----------------------------------------------------------------------
Ran 1 tests in 0.000s
OK

• 运行测试用例时，每完成一个单元测试， Python都打印一个字符：测试通过时打印一个句点；测试引发错误时打印一个E；测试导致断言失败时打印一个F。

添加新测试

• 如下

import unittest
from name_func import get_formatted_name

class NamesTestCase(unittest.TestCase):
    """测试name_func.py"""
    def test_first_last_name(self):
        """能够正确地处理像Janis Joplin这样的姓名吗？ """
        formatted_name = get_formatted_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()

• 同样，新添加的方法必须以test_打头，这个方法测试函数的另一个行为，即处理带中间名的名字。
• 测试成功，将输出如下

..
----------------------------------------------------------------------
Ran  2 tests in 0.000s
OK

测试类

各种断言方法

• Python在unittest.TestCase类中提供了很多断言方法，断言方法检查你认为应该满足的条件是否满足。
• 下表描述了6个常用的断言方法

方法	用途
assertEqual(a, b)	核实a == b
assertNotEqual(a, b)	核实a != b
assertTrue(x)	核实x为True
assertFalse(x)	核实x为False
assertIn(item, list)	核实item在list中
assertNotIn(item, list)	核实item不在list中

准备要测试的类

• 类的测试与函数的测试相似，因为你所做的大部分工作都是测试类中方法的行为，但也有一些不同之处。
• 以下是一个要被测试的类

# survey.py
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 responses:
        print('- ' + response)

测试AnonymousSurvey类

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

#test_survey.py
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()

方法setUp()

• 示例代码如下

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

• 方法setUp()做了两件事：创建一个调查对象；创建一个答案列表 。存储这两样东西的变量名包含前缀self，因此可在这个类的任何地方使用。
• 方法 test_store_three_response() 核 实 self.responses 中 的 第 一 个 答 案 self.responses[0] 被妥善地存储 。
• 方 法 test_store_three_response() 核实 self.responses 中的全部三个答案都被妥善地存储。