Fix compatibility of check_copy_ast_without_context with Py 3.13b1

Resolves: #16
alexmojaki · May 15, 2024 · 42e8a1f · 42e8a1f
1 parent 3d1564d
commit 42e8a1f
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -58,7 +58,16 @@ def check_copy_ast_without_context(tree):
     dump1 = ast.dump(tree)
     dump2 = ast.dump(tree2)
     normalised_dump1 = re.sub(
-        r", ctx=(Load|Store|Del)\(\)",
+        # Two possible matches:
+        # - first one like ", ctx=…" where ", " should be removed
+        # - second one like "(ctx=…" where "(" should be kept
+        (
+            r"("
+                r", ctx=(Load|Store|Del)\(\)"
+            r"|"
+                r"(?<=\()ctx=(Load|Store|Del)\(\)"
+            r")"
+        ),
         "",
         dump1
     )