Python Unittest简明教程

1 概述

单元测试是一种软件测试方法，通过来测试源代码中的各个单元，例如类，方法等，以确定它们是否符合要求。直观上来说，可以将单元视为最小的可测试部分。单元测试是程序员在开发过程中创建的短代码片段。 它构成了组件测试的基础。

2 unittest

unitest是python内置的用于测试代码的模块，它支持测试自动化，配置共享和关机代码测试。支持将测试样例聚合到测试集中，并将测试与报告框架独立。

为了实现这些，unittest 通过面向对象的方式支持了一些重要的概念。

1. 测试脚手架（test fixture）：表示为了开展一项或多项测试所需要进行的准备工作，以及所有相关的清理操作。举个例子，这可能包含创建临时或代理的数据库、目录，再或者启动一个服务器进程。
2. 测试用例（test case）：一个测试用例是一个独立的测试单元。它检查输入特定的数据时的响应。 unittest
   提供一个基类：TestCase ，用于新建测试用例。
3. 测试套件（test suite）： 是一系列的测试用例，或测试套件，或两者皆有。它用于归档需要一起执行的测试。
4. 测试运行器（test
   runner）：是一个用于执行和输出测试结果的组件。这个运行器可能使用图形接口、文本接口，或返回一个特定的值表示运行测试的结果。

2.1 创建一个简单的单元测试：

步骤：

1. 在程序中导入unittest模块。

2. 定义一个需要测试的函数。在下面的例子中，add()函数将会被测试。

3. 通过集成类unittest.TestCase创建一个测试用例类。

4. 在类中定义一个测试方法，该方法以test作为前缀。

5. 每个测试方法中都调用TestCase类的断言方法。里面有多种类型的断言。

6. assertEquals()比较把add()函数的结果与第二个参数做比较。如果不想等，会抛出异常assertionError。

7. 调用unittest模块中的main方法。

代码：（文件名为unittest01.py）

import unittest

def add(a, b):
    return a+b
    
# 一个类是一个测试用例，测试指类中的测试方法，测试用力中可以有多个测试。
class SimpleTest(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(3, 7), 10)

    def test_add2(self):
        self.assertEqual(add(3, 5), 10)

# __name__是一个全局变量，它的值是模块的名称。该判断语句表示只运行当前文件中的代码。
if __name__ == "__main__":
    unittest.main()

结果：

.F
======================================================================
FAIL: test_add2 (__main__.SimpleTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/home/hookind/Desktop/workspace/pythonTest/unittest01.py", line 13, in test_add2
    self.assertEqual(add(3, 5), 10)
AssertionError: 8 != 10

----------------------------------------------------------------------
Ran 2 tests in 0.001s

FAILED (failures=1)

2.2 命令行界面：

可以在命令行使用unittest模块运行模块、类和独立测试方法中的测试（这里测试指指测试用例类中的测试方法）:

python -m unittest test1
python -m unittest test_module.TestClass
python -m unittest test_module.TestClass.test_method

示例：对于上面的代码，可以通过下面这些命令运行。

python -m unittest unittest01
python3 -m unittest unittest01.SimpleTest
python3 -m unittest unittest01.SimpleTest.test_add

unittest支持一些命令行选项，可以通过下面的命令查看所有选项。

python3 -m unittest -h

3 API

这章只描述unittest模块中的类与方法。

3.1 TestCase类

该类的对象表示最小的可测试单元。

在TestCase类中的一些方法。

编号	方法	描述
1	setUp()	用来准备测试脚手架的方法，在调用测试方法之前，该方法会被立即调用
2	tearDown()	测试方法被调用后，该方法会被立即调用，并且结果也会被记录。即使测试方法抛出异常，该方法也会被调用
3	setUpClass()	一个类方法，在类中所有的测试运行前被调用
4	tearDownClass()	一个类方法，在类中所有的测试运行后被调用

脚手架（Fixture）
在测试用例类中，可以有大量的测试。这些方法可能需要初始化数据库连接、临时文件或其它的资源。这些就是脚手架。测试用力需要一些特别的方法配置和清理测试需要的脚手架。为了配置脚手架，覆盖setUp()方法；为了清理，覆盖tearDown()方法。

类脚手架（Class Fixture）
TestCase类有一个setUpClass()方法，该方法可以在TestCase类中的测试执行之前被覆盖以执行。同样，tearDownClass() 方法将在类中的所有测试之后执行。这两种方法都是类方法。因此，它们必须@classmethod注解装饰。

示例：

import unittest


def add(a, b):
    return a+b


class SimpleTest(unittest.TestCase):

    @classmethod
    def setUpClass(cls) -> None:
        print("==========setUpClass=========")
        return super().setUpClass()

    @classmethod
    def tearDownClass(cls) -> None:
        print("\n==========tearDownClass=========")
        return super().tearDownClass()

    def setUp(self):
        print("\n==========setUp=========")
        return super().setUp()

    def tearDown(self)