单元测试 python

单元测试是软件开发中的一种测试方法，用于验证代码中的各个独立单元（函数、方法等）的正确性。在Python中，可以使用内置的unittest模块来进行单元测试。

下面是一个简单的示例：

import unittest

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

class TestAdd(unittest.TestCase):

    def test_add(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.TestCase`创建了一个名为`TestAdd`的测试类。在`TestAdd`类中，我们定义了一个名为`test_add`的测试方法，使用`self.assertEqual`断言方法来断言函数的输出是否与预期结果相等。

然后，通过执行`unittest.main()`来运行单元测试。运行后，将会输出测试结果。

你可以根据需要编写更多的测试方法，并使用不同的断言方法来进行验证。通过这样的单元测试，你可以确保代码在各种情况下的正确性。