Python测试驱动开发：TDD实战教程

在软件开发领域，测试驱动开发（TDD）是一种备受推崇的实践方法。它不仅能够提高代码质量，还能增强开发者的信心。本文将带你深入了解Python中的TDD实战技巧，帮助你成为一名更优秀的开发者。

什么是TDD？

测试驱动开发（TDD）是一种迭代开发过程，其核心思想是“先写测试，再写代码”。具体步骤如下：

编写测试用例：在编写任何功能代码之前，先编写一个失败的测试用例。
运行测试：确保测试用例失败，验证测试的有效性。
编写最小代码：编写刚好能让测试通过的最少代码。
重构代码：在确保测试通过的前提下，优化代码结构。
重复上述步骤：不断迭代，直到完成所有功能。

Python中的TDD工具

在Python中，常用的测试框架有unittest、pytest等。本文将以pytest为例，展示TDD的实战应用。

安装pytest

首先，确保你已经安装了pytest。可以通过以下命令进行安装：

pip install pytest

实战案例：计算器程序

假设我们需要开发一个简单的计算器程序，支持加、减、乘、除四种基本运算。

Step 1: 编写测试用例

在项目根目录下创建一个名为test_calculator.py的文件，并编写第一个测试用例：

def test_add():
    result = add(1, 2)
    assert result == 3

此时，运行pytest，你会看到一个错误提示，因为add函数尚未定义。

Step 2: 编写最小代码

在项目根目录下创建一个名为calculator.py的文件，定义add函数：

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

再次运行pytest，测试应该通过。

Step 3: 添加更多测试用例

继续在test_calculator.py中添加更多测试用例：

def test_subtract():
    result = subtract(5, 3)
    assert result == 2

def test_multiply():
    result = multiply(4, 2)
    assert result == 8

def test_divide():
    result = divide(8, 2)
    assert result == 4

运行pytest，你会看到新的错误提示。

Step 4: 完善功能代码

在calculator.py中添加剩余的函数：

def subtract(a, b):
    return a - b

def multiply(a, b):
    return a * b

def divide(a, b):
    if b == 0:
        raise ValueError("Cannot divide by zero")
    return a / b

再次运行pytest，所有测试应该通过。

Step 5: 重构代码

在确保所有测试通过的前提下，可以对代码进行重构，例如添加类型注解、优化异常处理等。

def add(a: float, b: float) -> float:
    return a + b

def subtract(a: float, b: float) -> float:
    return a - b

def multiply(a: float, b: float) -> float:
    return a * b

def divide(a: float, b: float) -> float:
    if b == 0:
        raise ValueError("Cannot divide by zero")
    return a / b