1. 对函数测试

    Python标准库中的模块 unittest提供了代码测试工具，单元测试用于核实函数的某个方面没有问题，测试用例是一组单元测试，这些单元测试一起核实函数在各种情形下的行为都符合要求。

    要测试的函数：

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    """接受名和姓并返回整洁的姓名."""
     
     
    def get_formatted_name(first, last, middle=''):
        """Generate a neatly formatted full name."""
        if middle:
            full_name = first + ' ' + middle + ' ' + last
        else:
            full_name = first + ' ' + last
     
        return full_name.title()

    测试代码：

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    """测试函数."""
     
    from name_function import get_formatted_name
    import unittest
     
     
    class NamesTestCase(unittest.TestCase):
        """测试name_function.py."""
     
        def test_first_last_name(self):
            """是否能够正确处理姓名."""
            formatted_name = get_formatted_name('janis', 'joplin')
            self.assertEqual(formatted_name, 'Janis Joplin')
     
        def test_first_last_middle_name(self):
            """是否能够正确处理包含中间名的姓名."""
            formatted_name = get_formatted_name('wolfgang', 'mozart', 'amadeus')
            self.assertEqual(formatted_name, 'Wolfgang Amadeus Mozart')
     
    unittest.main()

    要为函数编写测试用例，可先导入模块 unittest 以及要测试的函数，再创建一个继承 unittest.TestCase 的类，并编写一系列方法对函数行为的不同方面进行测试。在这里我们使用了 unittest 类最有用的功能之一：一个断言方法—— assertEqual() ，该断言方法用来核实得到的结果是否与期望的结果一致。

    unittest Module中的断言方法

    • 代码行 unittest.main() 让Python运行这个文件中的测试。
    • 测试未通过时，不要修改测试，而应修复导致测试不能通过的代码。
    • 方法名必须以 test_ 打头，这样它才会在运行测试时自动运行。

  2. 对类测试

    需要进行测试的类：

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    """一个帮助管理匿名调查的类."""
     
     
    class AnonymousSurvey(object):
        """收集匿名调查问卷的答案."""
     
        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)

    对类进行测试的代码：

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    """author:王小平.
     
    date: 28/09/2016
    """
     
    from survey import AnonymousSurvey
    import unittest
     
     
    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('English')
            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()

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

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

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

  3. 运行测试用例时，每完成一个单元测试，Python 都打印一个字符：测试通过时打印一个句点；测试引发错误时打印一个 E；测试导致断言失败时打印一个 F。这就是你运行测试用例时，在输出的第一行中看到的句点和字符数量各不相同的原因。如果测试用例包含很多单元测试，需要运行很长时间，就可通过观察这些结果来获悉有多少个测试通过了。