要提升Gemini生成规范单元测试代码的质量，需在Prompt中显式约束语言框架、用例模式、代码风格、参考样例及分步验证五方面要求。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜
如果您希望 Gemini 生成结构清晰、符合主流编程语言规范的单元测试代码，但输出结果常出现命名随意、断言缺失或覆盖不足等问题，则可能是由于 Prompt 缺乏对测试规范的显式约束。以下是实现该目标的具体方法：
一、明确指定语言与框架规范
Gemini 的输出高度依赖 Prompt 中对技术上下文的精确描述。若未声明目标语言、测试框架及风格指南（如 pytest 的 assert 方式、JUnit 的 @Test 注解、或 Google Test 的 EXPECT_EQ），模型将默认采用通用模板，易偏离实际工程要求。
1、在 Prompt 开头声明目标环境，例如：“请使用 Python 3.9+ 和 pytest 7.x 框架”。
2、补充关键规范约束，例如：“所有测试函数名必须以 test_ 开头，使用小写字母和下划线；每个测试仅验证一个行为；断言必须使用 assert 语句而非 print 或 logging”。
3、提供待测函数签名作为上下文，例如：“被测函数为 def calculate_discount(price: float, rate: float) -> float:”。
二、嵌入典型测试用例模式
单纯要求“生成单元测试”无法触发 Gemini 对边界值、异常路径等关键场景的覆盖意识。需在 Prompt 中内嵌结构化用例模板，引导其按模式生成。
1、明确定义三类必含用例：“请分别生成：正常输入用例（如 price=100.0, rate=0.1）、边界值用例（如 price=0.0, rate=1.0）、非法输入用例（如 price=-10.0 或 rate=1.5）”。
2、为每类用例指定断言目标，例如：“非法输入用例中，必须抛出 ValueError，并使用 with pytest.raises(ValueError): 验证”。
3、要求为每个测试添加简短英文 docstring，说明其验证意图，例如：“每个测试函数开头需有单行 docstring，格式为 'Tests X when Y'”。
三、强制注入代码风格检查项
编程规范不仅涉及逻辑正确性，还包括可读性与可维护性。Gemini 不会自发遵循 PEP 8、Google Java Style 等规则，需通过 Prompt 显式注入检查点。
1、限定命名长度与格式，例如：“变量名不得缩写，test 函数参数名须为 price_input 和 rate_input，不可简写为 p 或 r”。
使用AIsa生成图像与视频。仅需一个API密钥即可调用Gemini 3 Pro Image（图像）和Qwen Wan 2.6（视频）。
2、规定空行与缩进行为，例如：“测试函数之间空一行；assert 语句前不加空格，后跟一个空格，如 assert result == 90.0”。
3、禁用不安全实践，例如：“禁止使用 time.sleep()、os.system() 或任何外部 I/O；所有测试必须纯内存执行”。
四、提供带注释的参考样例
Gemini 对少样本学习（few-shot learning）响应敏感。提供一段人工编写的、完全合规的测试样例，能显著提升其输出一致性与规范性。
1、在 Prompt 中插入完整样例，例如：“参考以下合规示例：def test_calculate_discount_normal_case():\n \"\"\"Tests discount calculation when price and rate are valid.\"\"\"\n result = calculate_discount(200.0, 0.15)\n assert result == 170.0”。
2、在样例后立即强调其要素，例如：“注意：该样例包含完整 docstring、无缩写参数名、单一 assert、无空行冗余、数值使用浮点字面量”。
3、要求新生成内容严格对齐该样例的缩进层级、标点风格与换行位置，例如：“新测试函数的 def 行、docstring 行、assert 行必须与样例保持相同缩进和空格数”。
五、分步验证式 Prompt 结构
将生成任务拆解为可验证的子指令，避免 Gemini 一次性输出模糊结果。每步输出均可独立校验是否满足规范。
1、第一步指令：“仅输出待测函数的全部可能输入分类，每类占一行，格式为：[分类名]：说明，例如 [正常输入]：price > 0 且 0 ≤ rate ≤ 1”。
2、第二步指令：“针对上一步中的 [边界值] 类，生成一个完整测试函数，包含 docstring、调用和 assert，不输出其他任何内容”。
3、第三步指令：“检查上一函数：是否含空行？docstring 是否以 Tests 开头？assert 是否唯一且无多余空格？仅回答‘通过’或列出不合规项”。
