用 AI 写单元测试是把函数代码和需求说明交给 AI，让它分析输入输出和边界情况，生成测试用例和测试代码。它能帮你快速补齐测试覆盖，也能帮你发现代码中没有考虑到的情况。读完这篇，你就能把它用到自己的日常工作里。

AI 写单元测试是什么，为什么有效

简单说，写测试需要系统地思考各种输入情况，这正是 AI 擅长列举的部分；你再根据业务判断取舍，就能更快地得到质量较高的测试。与手写测试相比，手写测试更贴合你对业务的理解，但列举边界情况耗时；AI 写测试能快速列出大量用例，你再根据业务取舍。

AI 写测试的意外收获，往往是发现代码本身的问题。当它列出“如果输入为空会怎样”“如果两个参数相等会怎样”时，你可能会发现自己从没考虑过这些情况。所以用例清单阶段比生成代码阶段更有价值：认真看清单，才能真正提升代码质量。

哪些情况值得用

  • 为已有代码补充测试。
  • 新功能开发时同步编写测试。
  • 重构前需要用测试保护现有行为。
  • 需要发现代码中未处理的边界情况。

怎么用：6 步上手

  1. 提供函数代码、使用的测试框架和项目的测试风格。
  2. 说明函数的预期行为和业务规则。
  3. 让 AI 先列出测试用例清单：正常情况、边界值、异常输入。
  4. 确认用例清单后再生成测试代码。
  5. 运行测试，对失败的用例判断是代码问题还是测试问题。
  6. 把测试纳入持续集成，定期维护。

可直接复制的提示词

下面以科研人员的工作场景为例，方括号里的内容换成你自己的信息即可：

请为下面这个函数编写单元测试，测试框架是[框架名]。
函数用途：处理科研人员工作中“实验组”相关的数据计算。
请：
1. 先列出测试用例清单：| 用例 | 输入 | 预期输出 | 类型（正常/边界/异常）|；
2. 至少包含：空输入、极大值、非法格式、重复数据等情况；
3. 等我确认清单后再写测试代码；
4. 测试命名清楚地表达测试意图。
函数代码：[粘贴]

效果示例

以下内容为虚构示例，仅用于说明效果：

示例（虚构）：一位开发者让 AI 为一个计算折扣的函数写测试。AI 列出的用例中包括“折扣叠加后价格为负数”的情况，运行测试后发现函数确实没有处理这种情况，于是补充了价格下限的逻辑。

新手最常犯的两个错误

  • 容易犯的错：不提供业务规则。更好的做法：说明函数的预期行为，AI 才能写出有意义的断言。
  • 容易犯的错：直接接受所有生成的测试。更好的做法：判断每个用例是否合理，删除重复和无意义的测试。

两个实用小技巧

  • 先让 AI 列用例清单，确认后再写代码，效率更高。
  • 让 AI 专门寻找“最可能出错的输入”

一个常见疑问

AI 写的测试能直接提交吗？ 需要运行并审查，确认测试有意义、断言正确，再按团队规范提交。

把 AI 接入日常工作流时，最重要的是明确哪一步由 AI 起草、哪一步由人把关。分工清楚，效率和质量才能兼顾。

下一步

入门之后，可以看看 AI 写单元测试的进阶技巧和常见错误，再结合 AI 编程助手一起使用。

相关阅读