Merge pull request #497 from python-rope/lieryan-496-matmul-op

Add MatMul operator to patchedast
python-rope · Jul 18, 2022 · 51b1bf8 · 51b1bf8
2 parents 79f2efe + 30cda85
commit 51b1bf8
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # **Upcoming release**
 
+## Bug fixes
+
+- #496,#497 Add MatMul operator to patchedast
+
 # Release 1.2.0
 
 Date: 2022-04-22

diff --git a/rope/refactor/patchedast.py b/rope/refactor/patchedast.py
@@ -277,6 +277,7 @@ def _child_nodes(self, nodes, separator):
         "Div": "/",
         "Mod": "%",
         "Pow": "**",
+        "MatMult": "@",
         "LShift": "<<",
         "RShift": ">>",
         "BitOr": "|",

diff --git a/ropetest/refactor/patchedasttest.py b/ropetest/refactor/patchedasttest.py
@@ -175,6 +175,12 @@ def test_and_node(self):
         checker.check_region("BoolOp", 0, len(source) - 1)
         checker.check_children("BoolOp", [NameConstant, " ", "and", " ", NameConstant])
 
+    def test_matmult_node(self):
+        source = "a @ b\n"
+        ast_frag = patchedast.get_patched_ast(source, True)
+        checker = _ResultChecker(self, ast_frag)
+        checker.check_children("BinOp", ["Name", " ", "@", " ", "Name"])
+
     def test_basic_closing_parens(self):
         source = "1 + (2)\n"
         ast_frag = patchedast.get_patched_ast(source, True)