python函数的测试

### 如何对Python函数进行单元测试最佳实践

编写高质量的软件离不开有效的测试。为了确保代码按预期工作，在开发过程中实施单元测试至关重要。

#### 使用unittest框架进行单元测试

`unittest` 是 Python 的标准库之一，提供了丰富的功能来支持单元测试[^1]。下面展示了一个简单的例子：

import unittest

def add(a, b):
    """返回两个数相加的结果"""
    return a + b


class TestAddFunction(unittest.TestCase):

    def test_add_positive_numbers(self):
        result = add(1, 2)
        self.assertEqual(result, 3)

    def test_add_negative_numbers(self):
        result = add(-1, -2)
        self.assertEqual(result, -3)


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

在这个案例中，定义了一个名为 `add()` 的简单函数，并创建了继承自 `unittest.TestCase` 类的测试类 `TestAddFunction` 来验证该函数的行为是否符合预期。

#### 测试驱动开发(TDD)

采用 TDD 方法意味着先编写失败的测试用例再实现相应的业务逻辑直到通过这些测试为止。这种方法有助于提高代码质量并减少错误的发生概率。

#### Mock对象的应用

有时候被测模块依赖外部资源或服务，这时可以利用 mock 技术模拟真实环境下的行为而不必实际调用它们。这不仅加快了测试速度也提高了稳定性。

from unittest.mock import patch

@patch('some_module.some_function')
def test_with_mock(mocked_func):
    mocked_func.return_value = "mock data"
    
    # 调用待测试的方法...