python基础篇(二十四)——测试代码(下)

待我称王封你为后i 2022-09-04 14:54 131阅读 1赞

## 前言 ##

我们上篇文章给大家介绍了[代码的测试][Link 1]；主要给大家介绍了函数的测试，主要包括单元测试以及可通过的测试以及当我们遇到测试通不过的时候我们应该如何修改代码，让测试类通过。本文给大家介绍测试类；首先给大家介绍测试类中各种断言的方法；

## 测试类 ##

上篇文章针对的是单个函数的测试；我们的本文是主要针对编写的类进行测试。在很多程序中都会用到类，因此，能够证明你的类能够正确的工作会大有裨益。如果针对类的测试通过了，你就能确信对类所做的改进没有意外地破坏其原有的行为。

## 1、各种断言方法 ##

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

## 2、一个要测试的类 ##

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

class AnonymousSurvery():
        """收集匿名调查问卷的答案"""
        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("Suvery result:")
            for response in self.responses:
                print('- ' + response)

这个类首先存储了一个你指定的调查问题，并创建了一个空列表，用于存储答案。这个类包含打印调查问题的方法、在答案列表中添加新答案的方法以及将存储在列表中的答案都打印出来的方法。要创建这个类的实例，只需提供一个问题即可。有了表示调查的实例后，就可使用`show_question()`来显示其中的问题，可使用`store_response()`来存储答案，并使用`show_results()`来显示调查结果。  
  为证明`AnonymousSurvey类`能够正确地工作，我们来编写一个使用它的程序：

from survey import AnonymousSurvery
    # 定义一个问题，并创建一个表示调查的AnonymousSurvery对象
    question = "What language did you first learn to speak?"
    my_survey = AnonymousSurvery(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()

这个程序定义了一个问题(“What language did you first learn to speak?”)，并使用这个问题创建了一个`AnonymousSurvey`对象。接下来，这个程序调用`show_question()`来显示问题，并提示用户输入答案。收到每个答案相同时将其存储起来。用户输入所有答案(输入q要求退出)后，调用`show_results()`来打印调查结果：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L09saXZlcmZseTE_size_16_color_FFFFFF_t_70_pic_center][]  
  `AnonymousSurvey类`可用于进行简单的匿名调查。假设我们将它放在了模块`survey`中，并想进行改进；让每位用户都可输入多个答案；编写一个方法，它只列出不同的答案，并指出了每个答案出现了多少次；再编写一个类，用于管理非匿名调查。  
  进行上述修改存在风险，可能会影响`AnonymousSurvey类`的当前行为。例如，允许每位用户输入多个答案时，可能不小心修改了处理单个答案的方式。要确认在开发这个模块时没有破坏既有行为，可以编写针对这个类的测试。

## 3、测试`AnonymousSurvey类` ##

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

我们首先导入了`模块unitest`以及要测试的`类AnonymouseSurvey`。我们将测试用例命名为`TestAnonymousSurvey`，它也继承了`unittest.TestCase`。第一个测试方法验证调查问题的单个答案被存储后，会包含在调查结果列表中。对于这个方法，一个不错的描述性名称是`tset_store_single)response()`。如果这个测试未通过，我们就能通过输出中的方法名词可知，在存储单个调查答案方面存在问题。  
  要测试类的行为，需要创建其实例，我们使用问题创建了一个名为`my_survey`的实例，然后使用方法`store_response()`存储了单个答案English。接下来，我们检查English是否包含在`列表my_survey_response`中，以核实这个答案是否被妥善地存储了。当我们运行`test_survey.py`时，测试通过了：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L09saXZlcmZseTE_size_16_color_FFFFFF_t_70_pic_center 1][]  
  这个当然很好，但只能收集一个答案的调查用途不大。下面来核实用户提供三个答案时，他们也将被妥善地存储。为此，我们在`TestAnonymousSurey`中再添加一些方法：

import unittest
    from survey import AnonymousSurvery
    class TestAnonymousSurvey(unittest.TestCase):
        """针对AnonymousSurvey类进行测试"""
        def test_store_single_response(self):
            """测试单个答案会被妥善地存储"""
            question = "What language did you first learn to speak?"
            my_survey = AnonymousSurvery(question)
            my_survey.store_response('English')
            self.assertIn('English', my_survey.responses)
        def test_store_three_response(self):
            """测试三个答案会被妥善地存储"""
            question = "What language did you first learn to speak?"
            my_survey = AnonymousSurvery(question)
            responses = ['Chinese', 'Rushion', 'English']
            for response in responses:
                my_survey.store_response(response)
            for response in responses:
                self.assertIn(response, my_survey.responses)
    unittest.main

我们将这个方法命名为`test_store_three_response()`，并像`test_store_single_response()`一样，在其中创建一个调查对象。我们定义了一个包含三个不同答案列表，再对其中每个答案都调用`store_response()`。存储这些答案后，我们使用一个循环来确认每个答案都包含在`my_response`中。当我们再次运行`test_survey.py`时，两个测试(针对单个答案的测试和针对三个答案的测试)都通过了：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L09saXZlcmZseTE_size_16_color_FFFFFF_t_70_pic_center 2][]  
.  前述做法的效果很好，但这些测试有些重复的地方。下面使用`unittest`的另一项功能来提高它们的效率。

## 4、方法`setUp()` ##

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

import unittest
    from survey import AnonymousSurvery
    class TestAnonymousSurvey(unittest.TestCase):
        """针对AnonymousSurvey类进行测试"""
        def setUp(self):
            """创建一个调查对象和一组答案，供使用的测试方法使用"""
            quetion =  "What language did you first learn to speak?"
            self.my_survey = AnonymousSurvery(quetion)
            self.responses = ['Chinese', 'Rushion', 'English']
        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_single_response()`核实`self.responses`中的第一个答案——`self.responses[0]`——被妥善地存储，而方法`test_store_three_responses()`核实`self.responses`中的全部三个答案都被妥善的存储。  
  `self.responses`再次运行`test_survey.py`时，这两个测试也都通过了：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L09saXZlcmZseTE_size_16_color_FFFFFF_t_70_pic_center 3][]  
  如果要扩展`AnonymousSurvey`，使其允许每位用户输入多个答案，这些测试将很有用。修改代码以接受多个答案后，可运行这些测试，确认存储单个答案或一系列答案的行为未受影响。  
  测试自己编写的类时，方法`setUp()`让测试编写起来更容易；可在`setUp()`方法中创建一系列实例并设置他们属性，再在测试方法中直接使用这些实例。相比于在每个测试方法中都创建实例并设置其属性，这要容易得多。  
  这里需要我们注意的是：运行测试用例时，每完成一个单元测试，Python都打印一个字符：测试通过时打印一个句点；测试引发错误时打印了一个E；测试导致断言失败时打印了一个F。这就是你运行测试用例时，在输出的第一行中看到的句点和字符数量各不相同的原因。如果测试用例包含很多单元测试，需要运行很长时间，就可以通过观察这些结果来获悉有多少个测试通过了。

## 总结 ##

我们上篇文章给大家介绍了[代码的测试][Link 1]；主要给大家介绍了函数的测试，主要包括单元测试以及可通过的测试以及当我们遇到测试通不过的时候我们应该如何修改代码，让测试类通过。本文给大家介绍测试类；主要包括各种断言的方法以及给了一个测试的案例，并且对其案例进行了测试，最后介绍了类测试中的`setUp()`方法。Python是一门注重实际操作的语言，它是众多编程语言中最简单，也是最好入门的。当你把这门语言学会了，再去学习java、go以及C语言就比较简单了。当然，Python也是一门热门语言，对于人工智能的实现有着很大的帮助，因此，值得大家花时间去学习。生命不息，奋斗不止，我们每天努力，好好学习，不断提高自己的能力，相信自己一定会学有所获。加油！！！

[Link 1]: https://blog.csdn.net/Oliverfly1/article/details/119696254
[415531e3e1824228b3bf363d60102f13.jpg_pic_center]: /images/20220829/933078b6d06e4b3ba89d18869563d8b3.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L09saXZlcmZseTE_size_16_color_FFFFFF_t_70_pic_center]: /images/20220829/a6196f5c7e21499dbb9a20c1fbbcee35.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L09saXZlcmZseTE_size_16_color_FFFFFF_t_70_pic_center 1]: /images/20220829/ab8988852e7c4eefa40b4162e73b3c64.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L09saXZlcmZseTE_size_16_color_FFFFFF_t_70_pic_center 2]: /images/20220829/42169f95748a485d8507620616f651c9.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L09saXZlcmZseTE_size_16_color_FFFFFF_t_70_pic_center 3]: /images/20220829/fe82fcc36ef444c59b0c84ffc31c1312.png