Fitten Code插件可自动为Python函数生成符合Pytest规范的测试脚本：启用后右键选择“Generate Test for Function”，自动创建test_*.py文件，基于函数签名和类型提示生成parametrize参数化测试及基础assert断言。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜
你需要为 Python 项目快速生成符合 Pytest 规范、能自动定位函数并构造基础断言的测试脚本，避免手动编写重复的 test_*.py 文件和冗余的 assert 语句。
安装并启用 Fitten Code 插件
在 VS Code 扩展市场中搜索 “Fitten Code”，点击安装 → 重启 VS Code → 按 Ctrl+Shift+P（Windows/Linux）或 Cmd+Shift+P（macOS）打开命令面板 → 输入 “Fitten: Enable” 并执行启用操作。【必须启用后才能触发测试生成】
插件图标会出现在左侧活动栏底部，鼠标悬停显示 “Fitten Code”。未启用时右键菜单中不会出现 “Generate Test” 选项。
为单个函数生成 Pytest 测试用例
打开待测 Python 文件（如 utils.py），将光标置于目标函数名正上方或函数内部任意位置 → 右键 → 选择 “Fitten: Generate Test for Function” → 插件自动分析函数签名、参数类型及 return 类型 → 在同目录下创建 test_utils.py（若不存在）或追加到已有 test_utils.py 中。
生成的测试函数名自动为 test_，使用 pytest.mark.parametrize 覆盖空参、典型值、边界值三类输入，并调用原函数后插入 assert result is not None。若函数含类型提示（如 def add(a: int, b: int) -> int:），则参数示例优先选用 int 类型值。
批量生成多个函数的测试脚本
方法一：按住 Ctrl（Windows/Linux）或 Cmd（macOS）多选多个函数定义行 → 右键 → 选择 “Fitten: Generate Tests for Selected Functions” → 插件逐个解析并合并写入同一 test_*.py 文件，函数间用空行分隔。
方法二：不选中任何代码 → 右键 → 选择 “Fitten: Generate Tests for All Functions in File” → 插件扫描当前文件全部 def 块（跳过 __init__、私有函数 _* 和异步函数 async def）→ 为每个公有函数生成独立测试用例。
【注意：该操作不可撤销，若 test_*.py 已存在且含手工编写的测试，新生成内容会直接追加到底部】
Fitten Code 1.0.3是一款由清华博士团队打造的AI编程助手，基于国产计图（Jittor）深度学习框架开发。它支持VS Code、JetBrains系列等主流IDE及80多种编程语言。核心功能包括智能代码补全、注释生成、代码解释、Bug检测、单元测试生成等，旨在全方位提升开发效率。该工具对个人用户免费开放。
自定义测试生成行为
按下 Ctrl+,（逗号）打开设置 → 搜索 “fitten test” → 修改以下三项：
• “Fitten Code › Test: Framework” 设为 pytest（默认即为此值）；
• “Fitten Code › Test: Assertion Style” 可选 assert_eq（生成 assert result == expected）、assert_is（生成 assert result is expected）或 no_assert（仅调用不加断言）；
• “Fitten Code › Test: Include Docstring Examples” 勾选后，若函数文档字符串含 >>> 示例，则自动提取为测试输入输出对。
修改后无需重启，下次生成立即生效。断言风格一旦设为 no_assert，生成的测试将不含任何 assert 行，需手动补全。
运行生成的测试脚本
确保项目根目录下有 pyproject.toml 或 pytest.ini 配置文件（最简配置只需包含 [tool:pytest] 或 [pytest] 标头）→ 终端进入项目根目录 → 执行 pytest test_*.py -v → 查看输出中每个 test_* 函数是否通过。
若报 ModuleNotFoundError，说明被测模块未在 PYTHONPATH 中；此时在终端先执行 export PYTHONPATH="${PYTHONPATH}:/your/project/root"（Linux/macOS）或 set PYTHONPATH=%PYTHONPATH%;C:\your\project\root（Windows），再运行 pytest。
