Sanitize lines

This commit is contained in:
Xinlong Hu
2025-08-04 23:50:59 -05:00
parent 7680bef29f
commit 3229cdab4f
+2 -2
View File
@@ -18,7 +18,7 @@ from rich.progress import Progress, BarColumn, TextColumn, TimeRemainingColumn,
from pylingual.control_flow_reconstruction.cfg import CFG from pylingual.control_flow_reconstruction.cfg import CFG
from pylingual.control_flow_reconstruction.structure import bc_to_cft from pylingual.control_flow_reconstruction.structure import bc_to_cft
from pylingual.main import print_result from pylingual.main import print_result
from pylingual.control_flow_reconstruction.source import SourceContext from pylingual.control_flow_reconstruction.source import SourceContext, sanitize_lines
from pylingual.editable_bytecode import PYCFile from pylingual.editable_bytecode import PYCFile
from pylingual.equivalence_check import TestResult, compare_pyc from pylingual.equivalence_check import TestResult, compare_pyc
from pylingual.utils.version import PythonVersion from pylingual.utils.version import PythonVersion
@@ -64,7 +64,7 @@ def run(file: Path, out_dir: Path, version: PythonVersion, print=False):
file = next(file.iterdir()) file = next(file.iterdir())
in_src = normalize_source(file.read_text(), replace_docstrings=True) in_src = normalize_source(file.read_text(), replace_docstrings=True)
src_lines = in_src.split("\n") src_lines = sanitize_lines(in_src.split("\n"))
in_path = out_dir / "a.py" in_path = out_dir / "a.py"
in_path.write_text(in_src, encoding="utf-8") in_path.write_text(in_src, encoding="utf-8")
in_pyc = out_dir / "a.pyc" in_pyc = out_dir / "a.pyc"