Copilot Chat：GitHub AI对话编程

import astimport inspectfrom typing import List, Dict, Optional, Anyfrom dataclasses import dataclass, field  @dataclassclass ParamInfo:    """函数参数信息"""    name: str    type_hint: str = "Any"    default: Optional[str] = None  @dataclassclass FunctionSignature:    """函数签名解析结果"""    name: str    params: List[ParamInfo] = field(default_factory=list)    return_type: str = "None"    docstring: str = ""    decorators: List[str] = field(default_factory=list)  class FunctionAnalyzer:    """函数签名分析器，解析Python函数并提取结构化信息"""     def __init__(self, source_code: str):        self.source_code = source_code        self.tree = ast.parse(source_code)        self.signatures: List[FunctionSignature] = []     def analyze(self) -> List[FunctionSignature]:        """分析源码中所有函数的签名信息"""        for node in ast.walk(self.tree):            if isinstance(node, ast.FunctionDef):                sig = self._extract_signature(node)                self.signatures.append(sig)        return self.signatures     def _extract_signature(self, node: ast.FunctionDef) -> FunctionSignature:        """从AST节点提取函数签名"""        sig = FunctionSignature(name=node.name)        sig.docstring = ast.get_docstring(node) or ""        sig.decorators = [            ast.dump(d) for d in node.decorator_list        ]        # 解析参数列表        args = node.args        type_comments = args.args        for i, arg in enumerate(args.args):            type_hint = "Any"            if arg.annotation:                type_hint = ast.unparse(arg.annotation)            default = None            defaults_offset = len(args.args) - len(args.defaults)            if i >= defaults_offset:                default = ast.unparse(args.defaults[i - defaults_offset])            sig.params.append(ParamInfo(                name=arg.arg,                type_hint=type_hint,                default=default            ))        # 解析返回类型        if node.returns:            sig.return_type = ast.unparse(node.returns)        return sig  class TestCaseGenerator:    """基于函数签名自动生成测试用例框架"""     def __init__(self, signatures: List[FunctionSignature]):        self.signatures = signatures     def generate_all(self) -> str:        """为所有函数生成测试用例框架"""        test_blocks = []        for sig in self.signatures:            test_blocks.append(self._generate_for_function(sig))        return "\n".join(test_blocks)     def _generate_for_function(self, sig: FunctionSignature) -> str:        """为单个函数生成测试用例"""        lines = []        lines.append(f"class Test{sig.name.title().replace('_', '')}:" )        lines.append(f'    """{sig.name} 的测试套件"""')        lines.append("")         # 正常路径测试        lines.append("    def test_normal_case(self):")        call_args = self._build_call_args(sig, "normal")        lines.append(f"        result = {sig.name}({call_args})")        lines.append("        assert result is not None")        lines.append("")         # 边界值测试        lines.append("    def test_boundary_values(self):")        call_args = self._build_call_args(sig, "boundary")        lines.append(f"        result = {sig.name}({call_args})")        lines.append("        # 验证边界条件下的行为")        lines.append("        assert result is not None")        lines.append("")         # 异常场景测试        lines.append("    def test_error_handling(self):")        call_args = self._build_call_args(sig, "error")        lines.append(f"        # 验证异常输入的处理")        lines.append(f"        result = {sig.name}({call_args})")        lines.append("        # 断言预期的异常或错误处理行为")        lines.append("")         return "\n".join(lines)     def _build_call_args(self, sig: FunctionSignature, scenario: str) -> str:        """根据测试场景构建调用参数"""        args = []        sample_values = {            "str": {"normal": "'hello'", "boundary": "''", "error": "None"},            "int": {"normal": "42", "boundary": "0", "error": "-1"},            "float": {"normal": "3.14", "boundary": "0.0", "error": "float('inf')"},            "list": {"normal": "[1, 2, 3]", "boundary": "[]", "error": "None"},            "dict": {"normal": "{'key': 'value'}", "boundary": "{}", "error": "None"},            "bool": {"normal": "True", "boundary": "False", "error": "None"},            "Any": {"normal": "'test_value'", "boundary": "None", "error": "object()"},        }        for param in sig.params:            if param.name == "self":                continue            type_key = param.type_hint            if type_key not in sample_values:                type_key = "Any"            args.append(sample_values[type_key][scenario])        return ", ".join(args)  def demo():    """演示：分析示例函数并生成测试用例"""    sample_code = '''def process_data(items: list, threshold: int = 10) -> dict:    """处理数据列表，过滤并统计"""    pass def calculate_average(values: list, precision: int = 2) -> float:    """计算平均值"""    pass'''    analyzer = FunctionAnalyzer(sample_code)    signatures = analyzer.analyze()    print("=== 函数签名分析结果 ===")    for sig in signatures:        print(f"函数: {sig.name}")        print(f"  参数: {[(p.name, p.type_hint, p.default) for p in sig.params]}")        print(f"  返回类型: {sig.return_type}")        print(f"  文档: {sig.docstring}")        print()    generator = TestCaseGenerator(signatures)    print("=== 生成的测试用例框架 ===")    print(generator.generate_all())  if __name__ == "__main__":    demo()