单元测试怎么写Python

单元测试是在软件开发过程中对程序最小可测试单元进行验证的过程。在Python中，通常使用unittest模块来进行单元测试。以下是编写Python单元测试的基本步骤：

1. **导入所需的库**：首先，你需要导入`unittest`库，这是Python的标准测试框架。

import unittest

2. **创建测试类**：创建一个继承自`unittest.TestCase`的类，每个方法都是一个测试用例。例如：

class MyFunctionTest(unittest.TestCase):
    #...

3. **编写测试方法**：每个方法前加上`test_`前缀，方法里执行被测试函数并检查结果。比如测试加法函数：

    def test_addition(self):
        result = add(3, 4)
        self.assertEqual(result, 7, "Expected 7 but got {}".format(result))

这里，`self.assertEqual`用于比较预期结果与实际结果是否相等。

4. **运行测试**：在主程序中，通过`unittest.main()`运行测试类或单独的测试方法：

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

5. **组织测试套件**：如果你有多个测试文件需要一起运行，可以将它们组合成一个测试套件。

6. **断言与错误处理**：除了`assertEqual`，还有其他断言方法如`assertTrue`、`assertFalse`、`assertIsNone`等，用于验证条件是否满足。

记得定期运行和更新测试，保持其覆盖率高，当修改代码时能够快速发现潜在的问题。