Simulator test fixes for Python 3.12 version upgrade

numba · Nov 23, 2023 · 66cb4d0 · 66cb4d0
1 parent 6d90d14
commit 66cb4d0
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/numba_rvsdg/tests/simulator.py b/numba_rvsdg/tests/simulator.py
@@ -10,6 +10,9 @@
 )
 
 import builtins
+import sys
+
+PYVERSION = sys.version_info[:2]
 
 
 class Simulator:
@@ -327,8 +330,11 @@ def op_FOR_ITER(self, inst):
         try:
             ind = next(tos)
         except StopIteration:
-            self.stack.pop()
             self.branch = True
+            if PYVERSION <= (3, 11):
+                self.stack.pop()
+            else:
+                self.stack.append(None)
         else:
             self.branch = False
             self.stack.append(ind)
@@ -427,3 +433,11 @@ def op_POP_JUMP_FORWARD_IF_NONE(self, inst):
     def op_POP_JUMP_BACKWARD_IF_NONE(self, inst):
         self.branch = self.stack[-1] is None
         self.stack.pop()
+
+    def op_END_FOR(self, inst):
+        self.stack.pop()
+        self.stack.pop()
+
+    def op_COPY(self, inst):
+        assert inst.argval > 0
+        self.stack.append(self.stack[-inst.argval])