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)  # 检查add(1, 2)是否等于3

    def test_add_negative_numbers(self):
        self.assertEqual(add(-1, -2), -3)  # 检查add(-1, -2)是否等于-3

    def test_add_mixed_numbers(self):
        self.assertEqual(add(-1, 1), 0)  # 检查add(-1, 1)是否等于0

# 运行测试
if __name__ == '__main__':
    unittest.main()

解释说明：

1. 导入unittest模块：unittest是Python自带的单元测试框架。
2. 定义被测试的函数：add(a, b)是一个简单的加法函数。
3. 创建测试类：TestAddFunction继承自unittest.TestCase，用于包含多个测试用例。
4. 编写测试用例：每个测试用例方法名必须以test开头，使用self.assertEqual()来验证函数的输出是否符合预期。
5. 运行测试：通过unittest.main()来运行所有测试用例。

如果你需要进一步了解如何编写和运行单元测试，可以参考Python官方文档或相关教程。