编写一段代码从输入的描述函数原型的字符串中获取函数的参数类型其中如果参数为函数标记参数类型为function;并给出测试用例
以下是Python代码示例:
import re
def get_arg_types(func_str):
# 使用正则表达式匹配函数参数部分
pattern = r"\((.*)\)"
match = re.search(pattern, func_str)
if not match:
return []
arg_str = match.group(1)
# 将参数字符串按逗号分隔为列表
arg_list = arg_str.split(",")
# 遍历参数列表,获取参数类型
arg_types = []
for arg in arg_list:
arg = arg.strip()
if arg == "":
continue
if arg.startswith("*"):
arg_types.append("pointer")
elif arg.startswith("&"):
arg_types.append("reference")
elif arg.startswith("function"):
arg_types.append("function")
else:
arg_types.append(arg)
return arg_types
# 测试用例
assert get_arg_types("int add(int a, int b)") == ["int", "int"]
assert get_arg_types("void (*signal(int sig, void (*func)(int)))(int)") == ["int", "function"]
assert get_arg_types("int *a, float &b") == ["pointer", "reference"]
以上代码中,我们使用了Python的re模块对函数原型字符串进行正则表达式匹配,获取到函数参数部分后,再根据逗号分隔,遍历参数列表获取参数类型。其中,我们将指针类型标记为"pointer",将引用类型标记为"reference",将函数类型标记为"function"。对于每个测试用例,我们都可以通过调用get_arg_types函数获取到对应的参数类型列表,并进行断言判断
原文地址: https://www.cveoy.top/t/topic/dblD 著作权归作者所有。请勿转载和采集!