python 写单元测试

Python的单元测试通常使用unittest模块来编写。下面是一个示例：

import unittest

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

class TestAddFunction(unittest.TestCase):
    def test_addition(self):
        self.assertEqual(add(2, 3), 5)
        self.assertEqual(add(-1, 1), 0)
        self.assertEqual(add(0, 0), 0)

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

在这个示例中，我们定义了一个简单的add函数，并使用unittest模块编写了一个名为TestAddFunction的测试类，其中包含一个名为test_addition的测试方法。在测试方法中，我们使用assertEqual断言来检查函数的输出是否与期望值相等。最后，我们使用unittest.main()来运行测试。

执行上述代码将输出以下结果：

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

这表示我们的测试通过了。如果测试失败，则会显示详细的错误信息。