| import unittest |
| from pathlib import Path |
| import tempfile |
|
|
| from agent_harness.repair_experiment import ( |
| RepairExperimentError, |
| extract_unified_diff, |
| oracle_hunk_ranges, |
| validate_patch_scope, |
| run_test_command, |
| ) |
|
|
|
|
| class RepairExperimentTests(unittest.TestCase): |
| def test_oracle_hunk_ranges_read_only_old_file_locations(self) -> None: |
| patch = ( |
| "diff --git a/x.go b/x.go\n--- a/x.go\n+++ b/x.go\n" |
| "@@ -55,6 +55,20 @@ type X struct {\n context\n" |
| ) |
|
|
| self.assertEqual(oracle_hunk_ranges(patch, "x.go"), ((55, 6),)) |
|
|
| def test_extracts_fenced_diff_without_rewriting_it(self) -> None: |
| response = { |
| "choices": [ |
| { |
| "message": { |
| "content": "```diff\ndiff --git a/x.go b/x.go\n--- a/x.go\n+++ b/x.go\n@@ -1 +1 @@\n-a\n+b\n```" |
| } |
| } |
| ] |
| } |
|
|
| patch = extract_unified_diff(response) |
|
|
| self.assertTrue(patch.startswith("diff --git a/x.go b/x.go")) |
| self.assertTrue(patch.endswith("+b\n")) |
|
|
| def test_rejects_modification_outside_evidence(self) -> None: |
| patch = "diff --git a/x.go b/x.go\n--- a/x.go\n+++ b/x.go\n@@ -1 +1 @@\n-a\n+b\n" |
|
|
| with self.assertRaises(RepairExperimentError): |
| validate_patch_scope(patch, ["y.go"]) |
|
|
| def test_accepts_modification_inside_evidence(self) -> None: |
| patch = "diff --git a/x.go b/x.go\n--- a/x.go\n+++ b/x.go\n@@ -1 +1 @@\n-a\n+b\n" |
|
|
| self.assertEqual(validate_patch_scope(patch, ["x.go"]), ("x.go",)) |
|
|
| def test_python_pytest_command_uses_frozen_interpreter(self) -> None: |
| with tempfile.TemporaryDirectory() as temporary: |
| tree = Path(temporary) |
| (tree / "test_smoke.py").write_text("def test_ok():\n assert True\n") |
| result = run_test_command(tree, "python -m pytest -q test_smoke.py") |
| self.assertEqual(result["returncode"], 0, result["stderr"]) |
|
|
