Use FileWrapper class in place of file where line number is required
diff --git a/tests/scripts/generate_code.py b/tests/scripts/generate_code.py
index b344f8c..bc44b8c 100644
--- a/tests/scripts/generate_code.py
+++ b/tests/scripts/generate_code.py
@@ -362,9 +362,8 @@
out.append(part)
part = ''
else:
+ part += str[i]
escape = not escape and str[i] == '\\'
- if not escape:
- part += str[i]
if len(part):
out.append(part)
return out
@@ -582,7 +581,7 @@
out_data_file.replace('\\', '\\\\')) # escape '\'
# Function code
- with open(funcs_file, 'r') as funcs_f, open(data_file, 'r') as data_f, open(out_data_file, 'w') as out_data_f:
+ with FileWrapper(funcs_file) as funcs_f, open(data_file, 'r') as data_f, open(out_data_file, 'w') as out_data_f:
suite_deps, dispatch_code, func_code, func_info = parse_functions(funcs_f)
snippets['functions_code'] = func_code
snippets['dispatch_code'] = dispatch_code
