python入门（七） 测试代码

单元测试和测试用例

Python标准库中的模块 unittest 提供了代码测试工具.

单元测试：核实函数某个方面没有问题

测试用例：一组单元测试，这些单元测试一起核实函数在各种情形下都符合要求

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

下面给出一个测试代码的例子：

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

unittest.main()

其中用到了unittest类最有用的功能之一：一个 断言 方法，断言方法用于核实得到的结果是否与期望的结果一致

unittest.main()会自动发现并运行当前模块中的所有测试单元，这样子希望添加新测试的时候，只需要在测试函数中添加新的方法即可

测试类

unittest.TestCase类中提供了多种断言方法

 unittest.TestCase类包含了方法setUp()，这样用于测试的样例只需要创建一次，就可以用于其他所有的测试方法，当类中包含了setUp()的时候，Python将优先运行它

例如：

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 ，因此可以在类的各个地方使用