def update_class(cls):
"""
Update the info for a class
Returns:
None
"""
parsed = Parsed(cls)
parsed.parse()
if not parsed.requires_update:
return
filename = inspect.getfile(cls)
with open(filename, 'r+') as location:
lines = location.readlines()
first_line, last_line = get_source_lines(lines, cls)
with open(filename, 'r+') as location:
for index, line in enumerate(location.readlines()[first_line:last_line]):
if config['requirement_info'] in line:
test_info_index = index + first_line
break
append_json_info(filename, test_info_index, create_json_info())
def update_func(function):
"""
Update the info for a function
Args:
f (function): The function to update
Returns:
dict: function information
test_id: id of the function encountered
"""
parsed = Parsed(function)
parsed.parse()
info_dict = parsed.test_info
if not parsed.requires_update:
return info_dict
filename = inspect.getfile(function)
with open(filename, 'r+') as location:
for index, line in enumerate(location.readlines()):
# We have not reached the function yet
if index < function.__code__.co_firstlineno:
continue
if 'TEST INFO' in line:
test_info_index = index
break
new_json = create_json_info()
append_json_info(filename, test_info_index, new_json)
return new_json
