高质量Python单元测试需明确提示词、声明函数边界并隔离依赖：一、依签名与docstring生成覆盖典型值、边界值及异常路径的测试；二、用mock隔离外部依赖并验证调用；三、结合Hypothesis生成变异输入；四、依据类型注解构造精确输入并校验返回类型。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜
如果您使用CodeGeeX为Python函数生成单元测试，但产出的测试用例覆盖不全、断言缺失或无法通过，可能是由于提示词不明确、函数边界未显式声明或依赖未隔离。以下是实现高质量测试的多种方法：
一、基于函数签名与docstring反向生成测试用例
利用CodeGeeX对已有函数的签名和文档字符串进行语义解析，自动生成输入-输出对及对应断言，确保测试覆盖典型值、边界值和异常路径。
1、在函数上方添加符合Google或NumPy风格的完整docstring，明确描述参数类型、返回值、Raises异常及示例用法。
2、在CodeGeeX中输入提示词：“请为以下Python函数生成pytest测试用例，覆盖正常输入、空值、类型错误和边界条件，每个测试函数名以test_开头，使用assert验证返回值：”后粘贴函数代码。
立即学习“Python免费学习笔记（深入）”；
3、运行生成结果，检查是否包含assert isinstance(result, expected_type)及with pytest.raises(ExpectedException)等关键断言结构。
二、注入测试桩（Test Stub）与模拟（Mock）指令
当被测函数调用外部API、数据库或随机模块时，需在提示中强制要求CodeGeeX引入unittest.mock或pytest-mock，避免测试因环境依赖而失败。
1、在提示词中明确写入：“所有对外部模块（如requests、random、datetime）的调用必须使用mock.patch或MagicMock替换，并验证调用次数与参数。”
2、提供一个最小化可运行的mock模板作为上下文，例如：“@patch('requests.get') def test_fetch_data(mock_get): mock_get.return_value.json.return_value = {'id': 1}”。
3、生成后确认测试函数内存在mock_前缀的形参及mock_obj.assert_called_once_with(...)验证语句。
三、采用Property-Based Testing引导生成变异输入
借助Hypothesis库的特性，驱动CodeGeeX生成基于策略的随机输入测试，提升对隐含边界和类型组合的覆盖率。
1、在提示词中加入：“使用hypothesis.strategies生成参数，例如@given(st.integers(min_value=0, max_value=100))，并确保每个测试函数标注@given和@settings(max_examples=10)。”
2、要求CodeGeeX为每个给定策略补充反例断言，如“assert result >= 0 and result
3、验证生成代码是否导入了from hypothesis import given, settings和from hypothesis.strategies import ...。
四、强制执行测试隔离与资源清理
防止测试间状态污染，需在提示中指定fixture生命周期控制与teardown行为，尤其针对文件操作、全局变量修改等场景。
1、提示词中声明：“所有涉及临时文件创建的测试必须使用tmp_path fixture，并在测试末尾显式调用tmp_path.joinpath(...).unlink(missing_ok=True)。”
2、若函数修改全局状态（如修改sys.path），要求生成@pytest.fixture(scope="function", autouse=True)并在yield前后分别保存与恢复状态。
3、检查生成代码中是否存在tmp_path / "test_file.txt"类路径构造，以及yield之后的clear_cache()或restore_state()调用。
五、嵌入类型注解驱动的测试生成约束
当函数已标注type hints（如def calc(x: int, y: Optional[float]) -> List[str]:），可引导CodeGeeX依据mypy兼容类型生成精确输入输出校验。
1、在提示中强调：“严格依据函数签名中的类型注解构造输入数据，例如int类型不得传入float，Optional[float]需分别测试None与非None分支。”
2、要求每个测试用例末尾添加assert语句验证返回值类型，例如assert isinstance(actual, list) and all(isinstance(s, str) for s in actual)。
3、确认生成的测试中无type: ignore注释，且所有变量名与参数名保持一致（如x_input对应x参数）。
