Adds support for IS_OP opcode

Mentioned in: - zrax#190 - zrax#191 - zrax#195
00mjk · Oct 8, 2021 · 588f24c · 588f24c
1 parent fd69853
commit 588f24c
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 0 deletions.
diff --git a/ASTree.cpp b/ASTree.cpp
@@ -1104,6 +1104,17 @@ PycRef<ASTNode> BuildFromCode(PycRef<PycCode> code, PycModule* mod)
                 stack.push(new ASTBinary(left, right, ASTBinary::BIN_IP_MAT_MULTIPLY));
             }
             break;
+        case Pyc::IS_OP_A:
+            {
+                PycRef<ASTNode> right = stack.top();
+                stack.pop();
+                PycRef<ASTNode> left = stack.top();
+                stack.pop();
+                // The operand will be 0 for 'is' and 1 for 'is not'.
+                // Map those back to the appropriate values in ASTCompare::op_str()
+                stack.push(new ASTCompare(left, right, operand ? 9 : 8));
+            }
+            break;
         case Pyc::JUMP_IF_FALSE_A:
         case Pyc::JUMP_IF_TRUE_A:
         case Pyc::JUMP_IF_FALSE_OR_POP_A:

diff --git a/tests/compiled/is_op.3.9.pyc b/tests/compiled/is_op.3.9.pyc
diff --git a/tests/input/is_op.py b/tests/input/is_op.py
@@ -0,0 +1,9 @@
+a = 10
+b = 10
+
+if a is b:
+    pass
+
+if a is not b:
+    pass
+
diff --git a/tests/tokenized/is_op.txt b/tests/tokenized/is_op.txt
@@ -0,0 +1,9 @@
+a = 10 <EOL>
+b = 10 <EOL>
+if a is b : <EOL>
+<INDENT>
+pass <EOL>
+<OUTDENT>
+if a is not b : <EOL>
+<INDENT>
+pass <EOL>