测试提效之AI 辅助生成测试用例：具体案例与代码实现

AI 技术在自动化测试中的应用日益成熟，尤其是在测试用例生成方面，能够大幅提升测试工程师的效率。通过 AI 模型（如 OpenAI GPT 系列），我们可以根据代码逻辑、接口定义或需求描述，快速生成高质量的测试用例。

1. AI 辅助生成测试用例的场景

1.1 单元测试生成

根据代码逻辑，AI 可以自动生成单元测试，包括正常路径、边界条件和异常场景。

1.2 接口测试用例生成

通过接口的 API 文档或定义，AI 自动生成请求参数和预期响应。

1.3 功能测试用例生成

根据需求文档，AI 可以生成功能测试用例，包括操作步骤和预期结果。

2. 示例 1：根据函数代码生成单元测试

我们以一个简单的数学函数为例，展示如何使用 AI 自动生成测试用例。

函数代码

def calculate_discount(price, discount_percentage):
    """
    计算折扣后的价格。
    :param price: 原价
    :param discount_percentage: 折扣百分比 (0-100)
    :return: 折扣后的价格
    """
    if price < 0:
        raise ValueError("价格不能为负数")
    if not (0 <= discount_percentage <= 100):
        raise ValueError("折扣百分比必须在 0 到 100 之间")
    return price * (1 - discount_percentage / 100)

AI 自动生成单元测试

利用 OpenAI GPT 或类似模型，通过输入代码生成测试用例。

代码实现：调用 OpenAI API 自动生成测试用例

import openai

# 调用 OpenAI API
def generate_test_case(function_code):
    prompt = f"以下是一个 Python 函数代码，请为其生成单元测试用例：\n\n{
     function_code}\n\n测试用例："
    response = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[{
   "role": "user", "content": prompt}]
    )
    return response["choices"][0]["message"]["content"]

# 示例调用
if __name__ == "__main__":
    function_code = """
    def calculate_dis