Add GEN_START support (zrax#392)

* Apparently should be enough ? * Add `GEN_START` test * Add `GEN_START` test compiled * Add `GEN_START` test tokenized * Smaller test since only POP is needed. * Smaller test since only POP is needed. * Smaller test since only POP is needed. * dos2unix
zzz9328fb · Oct 20, 2023 · b32f231 · b32f231
1 parent d62c8ad
commit b32f231
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/ASTree.cpp b/ASTree.cpp
@@ -2419,6 +2419,9 @@ PycRef<ASTNode> BuildFromCode(PycRef<PycCode> code, PycModule* mod)
         case Pyc::PUSH_NULL:
             stack.push(nullptr);
             break;
+        case Pyc::GEN_START_A:
+            stack.pop();
+            break;
         default:
             fprintf(stderr, "Unsupported opcode: %s\n", Pyc::OpcodeName(opcode & 0xFF));
             cleanBuild = false;

diff --git a/tests/compiled/GEN_START.3.10.pyc b/tests/compiled/GEN_START.3.10.pyc
diff --git a/tests/input/GEN_START.py b/tests/input/GEN_START.py
@@ -0,0 +1,2 @@
+def GEN_START():
+    yield 0
diff --git a/tests/tokenized/GEN_START.txt b/tests/tokenized/GEN_START.txt
@@ -0,0 +1,3 @@
+def GEN_START ( ) : <EOL>
+<INDENT>
+yield 0 <EOL>