python写单元测试

写 Python 单元测试需要使用 Python 自带的 unittest 模块，以下是一个简单的示例：

import unittest

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

class TestAddFunction(unittest.TestCase):
    def test_add_positive_numbers(self):
        self.assertEqual(add(3, 4), 7)
    
    def test_add_negative_numbers(self):
        self.assertEqual(add(-3, -4), -7)
    
    def test_add_mixed_numbers(self):
        self.assertEqual(add(3, -4), -1)
    
    def test_add_strings(self):
        self.assertRaises(TypeError, add, '3', '4')

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

在这个例子中，我们定义了一个名为 `add` 的函数，然后创建了一个名为 `TestAddFunction` 的测试类，其中包含了四个测试方法。每个测试方法都使用 `assertEqual` 方法来断言函数的返回值是否符合预期。最后，我们使用 `unittest.main()` 来执行测试。