CodeGeeX 可一键为 Python 函数生成 Pytest 单元测试,覆盖边界值、异常路径和典型输入;支持注释指令控制参数化与输出路径,要求英文注释触发解析。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

你需要为刚写完的 Python 函数快速生成覆盖边界值、异常路径和典型输入的单元测试,避免手动漏写空字符串、None、负数等关键用例。
在 VS Code 中一键生成 Pytest 测试
打开含目标函数的 .py 文件,将光标置于 def 行任意位置(例如 def calculate_discount(price: float, discount_rate: float) -> float: 这一行)。
右键 → 选择 “CodeGeeX: Generate Unit Test”。
生成的测试会自动插入当前文件末尾,函数名为 test_calculate_discount,含 import pytest 和至少 3 组断言:正常浮点数、零值输入、折扣率超 100% 的异常场景。
注意:如果函数带类型提示但未安装 mypy 或 pyright,CodeGeeX 可能误判参数范围——此时需手动补一个 assert price >= 0 防止负价传入。
用注释指令精准控制测试风格
方法一:在函数上方紧邻处添加单行注释:# @codegeex generate pytest test --parametrize price,discount_rate。
方法二:在函数上方写 GoDoc 风格注释,例如:"""Calculate final price after discount.Args: price: must be >= 0 discount_rate: between 0.0 and 1.0 inclusiveRaises: ValueError: if price <br><code>"""
按下 Ctrl+Enter(Windows/Linux)或 Cmd+Enter(macOS)触发生成。
这一步必须确保注释与 def 之间无空行,否则 CodeGeeX 无法关联上下文——【空行会导致注释被忽略】。
通过本地 Codex 或 OpenClaw OAuth 凭证直接调用 ChatGPT/Codex Responses 的 image_generation 工具来生成或编辑光栅图像,然后保存
在 JetBrains IDE 中生成 Java 单元测试
第一步:进入 Settings → Tools → CodeGeeX,填入有效 API Key 或本地服务地址 http://127.0.0.1:8080。
第二步:在编辑器中选中完整函数定义(从 public static double calculateDiscount(...) 到右大括号闭合)。
第三步:右键 → “Generate Unit Test with CodeGeeX”。
生成结果默认采用 JUnit 5 风格,含 @Test 注解和 Assertions.assertEquals 断言;若原函数抛出 IllegalArgumentException,测试中会自动生成 @Test(expected = IllegalArgumentException.class) 分支。
命令行批量生成测试(CI/CD 场景)
执行 pip install codegeex-cli 安装工具。
运行命令:codegeex-cli test --file ./src/utils.py --function calculate_discount --framework pytest。
该命令会解析 AST 提取所有分支路径,包括 if price 和 <code>elif discount_rate > 1.0: 对应的异常测试用例,输出到 ./tests/test_utils.py。
如果源文件含中文注释或 docstring,CLI 会自动跳过解析——【必须用英文注释才能触发分支识别】。

















