diff --git a/test/fuzz_diff_parser.py b/test/fuzz_diff_parser.py index 83efe72..2c0fe5b 100644 --- a/test/fuzz_diff_parser.py +++ b/test/fuzz_diff_parser.py @@ -27,7 +27,6 @@ import sys import os import random import pickle -import difflib import parso from parso.utils import split_lines @@ -55,6 +54,14 @@ def find_python_files_in_tree(file_path): yield os.path.join(root, name) +def _print_copyable_lines(lines): + for line in lines: + line = repr(line)[1:-1] + if line.endswith(r'\n'): + line = line[:-2] + '\n' + print(line, end='') + + class LineReplacement: def __init__(self, line_nr, new_line): self._line_nr = line_nr @@ -147,9 +154,10 @@ class FileModification: if print_code: print('Original:') - print(code) - print('Modified:') - print(modified_code) + _print_copyable_lines(code_lines) + print('\nModified:') + _print_copyable_lines(modified_lines) + print() grammar.parse(code, diff_cache=True) grammar.parse(modified_code, diff_cache=True)