Python 单元测试基础脚本

单元测试的概念：单元测试是针对程序中最小可测试单元进行检查和验证的过程。在Python中，通常一个函数或方法就是一个测试单元。
unittest框架 ： Python自带了一个名为unittest的单元测试框架，它受JUnit启发，为开发者提供了丰富的功能来创建和运行测试用例。使用unittest框架可以定义测试类和测试方法，并且这些测试方法名必须以test开头。
TestCase类 ： unittest.TestCase是unittest模块中最基本的测试用例类。你通过继承这个类来创建新的测试用例。在这个类里，你可以定义多个测试方法，每个方法都是独立的测试点。
断言（Assertions） ：断言用来验证代码的行为是否符合预期。unittest提供了多种内置的断言方法，如assertEqual(a, b)、assertTrue(x)、assertFalse(x)等。
设置和清理（Setup and Teardown）：
- setUp()：在每个测试方法运行之前都会调用此方法，用于准备测试环境。
- tearDown()：在每个测试方法运行之后都会调用此方法，用于清理测试环境。
- setUpClass() 和 tearDownClass()：这两个类方法分别在所有测试开始前和结束后执行一次，适用于需要共享的资源初始化和清理。
测试套件（Test Suites） ：测试套件是测试用例的集合，可以将多个测试用例组合起来一起运行。你可以使用unittest.TestLoader().loadTestsFromTestCase()加载特定的测试类到套件中，或者使用addTests()方法添加更多的测试。
运行测试 ：可以通过unittest.main()直接运行测试文件中的所有测试用例，也可以通过命令行指定要运行的测试。
pytest框架 ：虽然unittest是一个强大的工具，但还有另一个流行的测试框架叫做pytest，它简化了测试编写过程，支持参数化测试、插件系统等高级特性。对于初学者来说，pytest可能更容易上手。
Mock对象 ：在某些情况下，你可能不想真正调用外部依赖（如数据库、网络服务），这时就可以使用unittest.mock库提供的mock对象来模拟这些依赖的行为。

下面是一个简单的例子，演示如何使用unittest创建一个测试用例：

复制代码

python

复制代码

import unittest

# 假设我们有一个待测试的函数
def add(a, b):
    return a + b

class TestAddFunction(unittest.TestCase):
    
    def setUp(self):
        # 在这里设置任何测试所需的初始状态
        pass
    
    def tearDown(self):
        # 在这里清理任何测试后的残留状态
        pass
    
    def test_add_positive_numbers(self):
        self.assertEqual(add(1, 2), 3)
    
    def test_add_negative_numbers(self):
        self.assertEqual(add(-1, -2), -3)

if __name__ == '__main__':
    unittest.main()

这段代码定义了一个测试类TestAddFunction，其中包含了两个测试方法，分别测试正数相加和负数相加的情况。如果你想要运行这段代码，只需要将其保存为.py文件然后执行即可。

执行结果