From 588f24c87210436f3551dc5808d773b0331d5c96 Mon Sep 17 00:00:00 2001
From: John Richards <john@pushpoplabs.com>
Date: Thu, 7 Oct 2021 20:29:18 -0400
Subject: [PATCH] Adds support for IS_OP opcode

Mentioned in:
- #190
- #191
- #195
---
 ASTree.cpp                   |  11 +++++++++++
 tests/compiled/is_op.3.9.pyc | Bin 0 -> 139 bytes
 tests/input/is_op.py         |   9 +++++++++
 tests/tokenized/is_op.txt    |   9 +++++++++
 4 files changed, 29 insertions(+)
 create mode 100644 tests/compiled/is_op.3.9.pyc
 create mode 100644 tests/input/is_op.py
 create mode 100644 tests/tokenized/is_op.txt

diff --git a/ASTree.cpp b/ASTree.cpp
index c92480e..d6ee5c6 100644
--- 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
new file mode 100644
index 0000000000000000000000000000000000000000..964fb30d17f91c0fd31b69b47ae5df6056fc1f49
GIT binary patch
literal 139
zcmYe~<>g`kf~o!SiMBxcF^Gc<n1CDyATE{x5-AK(Ak3J`kjhxfP$U4R8H*%R7=sx!
znO=fS^3!Cx#h7@DF=-`35etwHCVp{b7RToo=oM7n;;_lhPbtkwwPOSs$^j%;7+IJ&
Hm^l~${0A6v

literal 0
HcmV?d00001

diff --git a/tests/input/is_op.py b/tests/input/is_op.py
new file mode 100644
index 0000000..d7dcfbb
--- /dev/null
+++ 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
new file mode 100644
index 0000000..fead02e
--- /dev/null
+++ 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>