Python学习(六)----------单元测试

Python3之unittest单元测试框架

1. unittest 的定义

unittest 被称作 Python 版本的 JUnit ，有时也被叫做“PyUnit”。主要是用于 Python 语言程序的单元测试。

2. unittest 框架的 4 个重要概念

官方文档给出了 unittest 框架中 4 个重要的概念，介绍如下。

1）test fixture(测试固件)

一个 test fixture 代表一个或多个测试执行前的准备动作和测试结束后的清理动作 ，例如，创建数据库连接，启动服务进程，测试环境好清理或者关闭数据库连接等。

2）test case(测试用例)

一个 test case 就是一个最小测试单元，也就是一个完整的测试流程。针对一组特殊的输入进行特殊的验证与响应。通过继承 unittest 提供的测试基类(TestCase)，可以创建新的测试用例。

3）test suit(测套件)

一个 test suit 就是一组测试用例，一组测试套件或者两者共同组成的集合。它的作用是将测试用例集合到一起，然后一次性执行集合中所有的测试用例。

4）test runner(测试运行器)

一个 test runner 由执行设定的测试用例和将测试结果提供给用户两部分功能组成。

3. 单元测试加载方法

两种单元测试的加载方法：

1）直接通过 unittest.main() 方法加载单元测试的测试模块，测试方法执行顺序按照方法名的字符串的ASCII码升序排序。

2）将所有的单元测试用例(Test Case)添加到测试套件(Test Suite)集合中，然后一次性加载所有测试对象。

4. 测试用例

TestCase 作为 unittest 单元测试框架中测试单元的运行实体，单元测试脚本编写员可以通过他派生出自定义的测试过程和方法。TestCase 子类从父类继承的几个特殊方法，在测试用例执行时均会被一次执行，

TestCase 类中定义的几个特殊方法如下。

1）setUp( )：每个测试方法运行前运行，测试前的初始化工作。

2）tearDown( )：每个测试方法运行结束后运行，测试后的清理工作。

3）setUpClass( )：所有测试方法运行前运行，单元测试前期准备，必须使用 @classmethod 装饰器进行装修饰，setUp( )函数之前执行，整个测试过程只执行一 次。

4）tearDownClass( )：所有测试方法结束后运行，单元测试后期清理，必须使用 @classmethod 装饰器进行装修饰，tearDown( )函数之后执行，整个测试过程只执行一次。

 

 

测试函数

单元测试和测试用例

• 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 中的全部三个答案都被妥善地存储。