使用Claude Code编写Python单元测试：高效提示词设计与工程实践

1次阅读

没有评论

共计 1865 个字符，预计需要花费 5 分钟才能阅读完成。

在 Python 项目中编写单元测试时，开发者常遇到以下痛点：

重复代码多：为不同输入组合编写相似测试用例时，容易产生大量重复代码
边界条件遗漏：人工编写时容易忽略异常输入、极端值等边界情况
维护成本高：业务逻辑变更后，需要手动同步修改多个测试用例

传统解决方案如 pytest 参数化和 mock 库虽然能部分解决问题，但仍存在局限：

参数化需要手动设计测试数据，无法自动推断有效边界
mock 对象配置复杂，特别是涉及多层依赖时
测试覆盖率提升依赖开发者经验，新人难以快速上手

角色定义 ：明确要求 AI 扮演 ” 资深测试工程师 ” 角色
输入规范 ：必须包含待测函数签名和功能说明
输出要求 ：指定使用 pytest 框架，包含正常 / 异常场景

"""
角色：你是一位专注 Python 单元测试的工程师
任务：为以下函数生成 pytest 测试用例
要求：1. 使用 @pytest.mark.parametrize 覆盖边界条件
2. 对可能抛出的异常进行测试
3. 包含至少 3 个正常用例和 2 个异常用例
4. 每个断言添加注释说明验证逻辑

函数定义：{函数签名}
功能描述：{功能说明}
"""

def divide_numbers(dividend: float, divisor: float) -> float:
    """执行除法运算，当除数为 0 时抛出 ValueError"""
    if divisor == 0:
        raise ValueError("除数不能为 0")
    return dividend / divisor

import pytest

@pytest.mark.parametrize("a,b,expected", [(10, 2, 5),    # 正常整数除法
    (5.5, 2, 2.75), # 浮点数运算
    (0, 1, 0),      # 被除数为 0
])
def test_divide_normal_cases(a, b, expected):
    """验证各种合法输入的运算结果"""
    assert divide_numbers(a, b) == expected

@pytest.mark.parametrize("a,b", [(1, 0),         # 除数为 0
    ("1", 2),       # 类型错误
])
def test_divide_exceptional_cases(a, b):
    """验证异常场景的正确处理"""
    with pytest.raises((ValueError, TypeError)):
        divide_numbers(a, b)

GitHub Actions 配置示例：

name: Test with Claude

on: [push, pull_request]

jobs:
  test:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - name: Set up Python
      uses: actions/setup-python@v4
      with:
        python-version: '3.10'
    - name: Install dependencies
      run: |
        python -m pip install --upgrade pip
        pip install pytest pytest-cov
    - name: Generate tests
      env:
        CLAUDE_API_KEY: ${{secrets.CLAUDE_API_KEY}}
      run: python scripts/generate_tests.py
    - name: Run tests
      run: |
        pytest --cov=src --cov-report=xml
    - name: Upload coverage
      uses: codecov/codecov-action@v3