codegeex支持多平台自动生成单元测试:vs code一键生成、jetbrains插件调用、cli命令行工具及web ide在线生成,覆盖边界值、异常路径与典型场景。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

如果您希望为现有函数快速生成单元测试用例,CodeGeeX 可以基于函数签名与实现逻辑自动生成覆盖输入边界、异常路径和典型场景的测试代码。以下是利用 CodeGeeX 完成该任务的具体操作方式:
一、在 VS Code 中安装并启用 CodeGeeX 插件
CodeGeeX 提供官方 VS Code 扩展,支持对光标所在函数一键生成测试用例,前提是插件已正确配置并连接至后端服务。
1、打开 VS Code,进入扩展市场(Ctrl+Shift+X),搜索“CodeGeeX”。
2、点击“安装”,等待插件下载并启用。
3、首次启动时,按提示完成用户登录或匿名初始化,并确认模型服务可访问。
4、打开含目标函数的源文件,将光标置于函数定义行(如 def calculate_total(...) 或 function processData() {...})。
5、右键选择“CodeGeeX: Generate Unit Test”,或使用快捷键 Ctrl+Alt+T(Windows/Linux)/ Cmd+Option+T(macOS)。
二、在 JetBrains IDE 中通过插件调用 CodeGeeX 测试生成功能
JetBrains 系列 IDE(如 PyCharm、IntelliJ IDEA)可通过第三方插件桥接 CodeGeeX API,实现对 Java、Python、JavaScript 函数的测试用例生成。
1、进入 Settings → Plugins,点击 Marketplace,搜索“CodeGeeX AI Assistant”。
2、安装插件并重启 IDE。
3、确保在 Settings → Tools → CodeGeeX 中填写有效 API Key(如使用开源部署版,则填入本地服务地址 http://127.0.0.1:8080)。
4、在编辑器中选中目标函数全部代码块(包括函数头与主体),右键选择“Generate Unit Test with CodeGeeX”。
5、生成结果将自动插入当前文件下方或新测试文件中,默认采用与原函数所在框架一致的测试风格(如 pytest / JUnit 5 / Jest)。
CodeGeeX 2.21.0是智谱AI推出的AI编程助手版本,对Inline Chat功能进行了重大优化,包括UI设计升级、支持流式输出以提升响应速度,并新增“查看思路”按钮以便理解代码原理。同时,该版本在编辑器中新增CodeLens控件,支持一键“解释”代码或“添加注释”。目前该工具对个人开发者免费开放。
三、使用命令行工具 codegeex-cli 直接生成测试代码
对于无图形界面环境或 CI/CD 集成场景,codegeex-cli 支持从指定源码路径提取函数并输出标准格式测试用例,无需 IDE 依赖。
1、执行 pip install codegeex-cli 安装命令行工具。
2、运行 codegeex-cli test --file ./src/math_utils.py --function find_max 指定待测文件与函数名。
3、工具自动解析 AST 获取参数类型、返回值、docstring 及控制流分支点。
4、输出内容直接写入 ./tests/test_math_utils.py,包含 assert 断言、mock 调用桩及 @pytest.mark.parametrize 参数化用例。
5、若函数含外部依赖,工具会自动添加 patch 装饰器占位符,提示用户补充真实 mock 行为。
四、在 Web IDE 界面中粘贴代码后触发测试生成
CodeGeeX 官方网页版(https://codegeex.cn)提供零配置在线交互能力,适用于临时验证、教学演示或受限开发环境。
1、访问官网,登录账户后进入“Code Generation”标签页。
2、在左侧编辑框中粘贴完整函数定义(需含语言标识注释,例如 # language: python)。
3、在提示词输入框中键入:"生成覆盖正常输入、空输入、类型错误和边界值的单元测试,使用 pytest 格式"。
4、点击“Run”,等待模型响应,右侧将显示可直接复制的测试代码块。
5、生成结果中所有 assert 语句均附带中文注释说明预期行为,便于人工校验逻辑完整性。










