【合入来源】

• 需求
• 问题单
• issue/工单
• 重构优化
• 资料更新

关联 issue：https://gitcode.com/Ascend/pytorch/issues/1637

【修改方案】

本 PR 新增 torch.fx 代码生成相关 API 的单元测试，覆盖如下 API：

• torch.fx.Graph.python_code
• torch.fx.graph.PythonCode
• torch.fx.Graph.set_codegen

具体修改内容：

1. 新增 test/fx/test_fx_codegen_api.py 测试文件。
2. 增加 test_graph_python_code_returns_python_code 用例，验证 Graph.python_code("self") 返回对象类型为 PythonCode，并校验返回对象包含 src、globals 字段。
3. 增加 test_graph_set_codegen 用例，验证 Graph.set_codegen 设置自定义 CodeGen 后，GraphModule 可重新编译并正常执行。
4. 该组 API 属于 torch.fx Python 层代码生成接口，不涉及新增 NPU 算子实现。

【资料变更】

不涉及。

本 PR 仅新增测试用例，不修改资料文档。
torch.fx.graph.PythonCode 在 native_apis 文档中的支持状态已通过单独 docs PR 补充。

【接口变更】

不涉及。

本 PR 不修改客户可见接口，不修改 API 行为，仅新增测试用例验证现有 API 兼容性。

【功能验证】

测试场景：

1. 验证 torch.fx.Graph.python_code 可正常生成 PythonCode。
2. 验证 torch.fx.graph.PythonCode 对象包含 src 和 globals 字段。
3. 验证 torch.fx.Graph.set_codegen 设置自定义 CodeGen 后，GraphModule 可重新编译并正常执行。

测试命令：

TORCH_DEVICE_BACKEND_AUTOLOAD=0 python -u test/fx/test_fx_codegen_api.py -v

测试结果：

`test_graph_python_code_returns_python_code (__main__.TestFXCodegenAPI.test_graph_python_code_returns_python_code) ... ok`

`test_graph_set_codegen (__main__.TestFXCodegenAPI.test_graph_set_codegen) ... ok`

`Ran 2 tests in 0.168s`

`OK`

说明：

该 API 为 torch.fx Python 层代码生成接口，不涉及 NPU 算子计算。本次测试主要验证 API 功能行为与 PyTorch 原生保持一致。

# 【CheckList】

- [x] 代码注释完备，正确记录错误日志
- [x] 代码实现进行了返回值、空指针等校验
- [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等
- [ ] PR持续集成流水线（CI）执行通过，代码检查无异常