test_corpus.py: improve line-wise comparison

Author: elifarley

tests/test_corpus.py

@@ -93,13 +93,19 @@ def check_expected_files(dir_path: Path):
                 expected_file = test_dir / f"expected.{rel_path}"
                 assert expected_file.exists(), f"Expected file not found: {expected_file}"
 
-                expected_content = f"[{rel_path}] \n" + expected_file.read_text()
-                result_content = f"[{rel_path}] \n" + path.read_text()
-                assert result_content.strip() == expected_content.strip(), \
+                expected_content = file_to_lines(expected_file, rel_path)
+                actual_content = file_to_lines(path, rel_path)
+                assert actual_content == expected_content, \
                     f"Output does not match expected content for {rel_path}"
 
         check_expected_files(editor.root_path)
 
     except Exception:
         editor._failed = True  # Mark as failed to preserve temp directory
         raise
+
+
+def file_to_lines(file_path, rel_path):
+    expected_content = [f"#{i} [{rel_path}]{c}" for i, c in enumerate(file_path.read_text().splitlines())]
+    return expected_content
+