TODO Fix bug

Author: elifarley

src/cedarscript_editor/cedarscript_editor.py

@@ -311,7 +311,7 @@ def restrict_search_range(action, target, identifier_finder: IdentifierFinder) -
                                     return inner_boundaries.whole
                                 case RangeSpec() as inner_range_spec:
                                     return inner_range_spec
-                        case _ as invalid: # Marker (LINE) or Segment
+                        case _ as invalid: # Segment
                             # TODO
                             raise ValueError(f'Not implemented: {invalid}')
     return RangeSpec.EMPTY

tests/corpus/update.identifier.replace.line/chat.xml

@@ -4,7 +4,7 @@ UPDATE FUNCTION "fun2"
 FROM FILE "1.py"
 REPLACE LINE "def fun2(self, a, b, c):"
 WITH CONTENT '''
-µ@0:def fun2(a, b, c):
+@0:def fun2(a, b, c):
 ''';
 
 UPDATE FUNCTION "calculate"

tests/corpus/x.update.identifier.replace.segment/1.py

@@ -0,0 +1,60 @@
+class A:
+    def a(self):
+        def calculate(
+                a,
+                b,
+                c,
+                d,
+                e
+        ):
+            return a + b
+    def helper(
+            a,
+            b
+    ):
+        pass
+class B:
+    def helper0(
+            a,
+            b
+    ):
+        pass
+    def a(self):
+        def calculate(
+                a,
+                b,
+                c,
+                d,
+                e
+        ):
+            return a + b
+    def helper(
+            a,
+            b,
+            e
+    ):
+        pass
+def a(self):
+    def calculate(
+            a,
+            b,
+            c,
+            d,
+            e
+    ):
+        return a + b
+class C:
+    def a(self):
+        def calculate(
+                a,
+                b,
+                c,
+                d,
+                e
+        ):
+            return a + b
+    def helper(
+            a,
+            b
+    ):
+        pass

tests/corpus/x.update.identifier.replace.segment/chat.xml

@@ -0,0 +1,15 @@
+<no-train>
+```CEDARScript
+UPDATE CLASS "B"
+FROM FILE "1.py"
+REPLACE SEGMENT
+STARTING BEFORE LINE "a," OFFSET 1
+ENDING AFTER LINE "e" OFFSET 0
+WITH CONTENT '''
+@-2:def calculate_plus_5(
+@-1:a, b
+@-2):
+@-1:a += 5
+''';
+```
+</no-train>

tests/corpus/x.update.identifier.replace.segment/expected.1.py

@@ -0,0 +1,57 @@
+class A:
+    def a(self):
+        def calculate(
+                a,
+                b,
+                c,
+                d,
+                e
+        ):
+            return a + b
+    def helper(
+            a,
+            b
+    ):
+        pass
+class B:
+    def helper0(
+            a,
+            b
+    ):
+        pass
+    def a(self):
+        def calculate_plus_5(
+            a, b,
+        ):
+            a += 5
+            return a + b
+    def helper(
+            a,
+            b,
+            e
+    ):
+        pass
+def a(self):
+    def calculate(
+            a,
+            b,
+            c,
+            d,
+            e
+    ):
+        return a + b
+class C:
+    def a(self):
+        def calculate(
+                a,
+                b,
+                c,
+                d,
+                e
+        ):
+            return a + b
+    def helper(
+            a,
+            b
+    ):
+        pass

tests/test_corpus.py

@@ -78,8 +78,8 @@ 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 = expected_file.read_text()
-                    result_content = path.read_text()
+                    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(), \
                         f"Output does not match expected content for {rel_path}"