python 单元测试

import unittest

# 定义一个简单的函数，用于测试
def add(a, b):
    return a + b

# 创建一个测试类，继承自unittest.TestCase
class TestAddFunction(unittest.TestCase):
    # 测试用例方法必须以test开头
    def test_add_positive_numbers(self):
        self.assertEqual(add(1, 2), 3)  # 验证1 + 2 = 3

    def test_add_negative_numbers(self):
        self.assertEqual(add(-1, -2), -3)  # 验证-1 + (-2) = -3

    def test_add_zero(self):
        self.assertEqual(add(0, 0), 0)  # 验证0 + 0 = 0

# 如果该文件作为主程序运行，则执行测试
if __name__ == '__main__':
    unittest.main()

解释说明：

1. 导入模块：我们导入了unittest模块，这是Python标准库中的单元测试框架。
2. 定义被测试的函数：add(a, b)是一个简单的加法函数，返回两个数的和。
3. 创建测试类：TestAddFunction类继承自unittest.TestCase，表示这是一个测试类。
4. 编写测试用例：每个测试方法都以test_开头，并使用self.assertEqual()来验证函数的输出是否符合预期。
5. 运行测试：如果这个脚本是直接运行的（而不是被导入），则会调用unittest.main()来执行所有测试。

通过这种方式，你可以轻松地为你的代码编写单元测试，确保其行为符合预期。