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

1次阅读

没有评论

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

编写单元测试是保证代码质量的重要手段，但在实际开发中，我们常常面临以下痛点：

重复代码多 ：相似的测试用例需要反复编写模板代码
边界条件遗漏 ：人工编写容易忽略异常分支和边界情况
维护成本高 ：当业务代码变更时，测试用例需要同步修改

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

参数化需要手动设计测试数据，无法自动推断边界值
mock 对象配置复杂，特别是多层嵌套的对象
测试覆盖率提升依赖开发者的经验

Claude Code 通过智能提示词可以自动生成高质量的单元测试。以下是核心设计原则：

角色定义 ：明确要求 Claude 扮演 ” 资深测试工程师 ” 角色
输入输出规范 ：
提供完整的函数签名（包括参数和返回值类型）
说明函数的业务场景和关键逻辑
边界条件提示 ：
要求覆盖所有参数的可能取值边界
特别关注 None、空值、极值等特殊情况

 你是一位经验丰富的测试工程师，请为以下 Python 函数编写单元测试：函数功能：{清晰描述函数用途}
函数签名：{包含参数类型和返回类型}

要求：1. 使用 pytest 框架
2. 覆盖所有参数的有效和无效输入组合
3. 包含至少 3 个正常场景和 2 个异常场景
4. 对依赖的外部服务使用 pytest-mock 进行模拟
5. 为每个测试用例添加清晰的断言说明

def clean_user_data(data: dict) -> dict:
    """清洗用户数据，处理缺失值和格式转换"""
    result = data.copy()
    # 处理年龄字段
    if 'age' in result:
        try:
            result['age'] = int(result['age'])
        except (ValueError, TypeError):
            result['age'] = None
    # 处理姓名字段
    if 'name' in result:
        result['name'] = str(result['name']).strip()
    return result

import pytest

@pytest.mark.parametrize('input_data,expected', [
    # 正常场景
    ({'age': '25', 'name': 'Alice'}, {'age': 25, 'name': 'Alice'}),
    ({'age': 30, 'name': 'Bob'}, {'age': 30, 'name': 'Bob'}),
    # 边界场景
    ({'age': '','name':''}, {'age': None, 'name': ''}),
    ({'age': 'invalid', 'name': 123}, {'age': None, 'name': '123'}),
])
def test_clean_user_data(input_data, expected):
    """测试数据清洗功能"""
    assert clean_user_data(input_data) == expected

@pytest.fixture
def mock_external_service(mocker):
    """模拟外部服务"""
    return mocker.patch('module.external_api', return_value={'status': 'ok'})

GitHub Actions 配置示例：

name: Python CI

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.9'
    - name: Install dependencies
      run: |
        python -m pip install --upgrade pip
        pip install pytest pytest-cov
    - name: Generate tests with Claude
      run: python scripts/generate_tests.py
    - name: Run tests
      run: |
        pytest --cov=./ --cov-report=xml
    - name: Upload coverage
      uses: codecov/codecov-action@v3