generator expressions must also have nested code objects

Author: timfel

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/code/PCode.java

@@ -60,6 +60,7 @@
 import com.oracle.graal.python.nodes.frame.FrameSlotIDs;
 import com.oracle.graal.python.nodes.frame.GlobalNode;
 import com.oracle.graal.python.nodes.function.FunctionDefinitionNode;
+import com.oracle.graal.python.nodes.function.GeneratorExpressionNode;
 import com.oracle.graal.python.nodes.generator.GeneratorFunctionRootNode;
 import com.oracle.graal.python.nodes.literal.SimpleLiteralNode;
 import com.oracle.truffle.api.CompilerDirectives;
@@ -245,13 +246,12 @@ private static Object[] extractConstants(RootNode rootNode) {
             public boolean visit(Node node) {
                 if (node instanceof SimpleLiteralNode) {
                     constants.add(((SimpleLiteralNode) node).getValue());
-                    return false;
                 } else if (node instanceof FunctionDefinitionNode) {
                     constants.add(new PCode(PythonBuiltinClassType.PCode, ((FunctionDefinitionNode) node).getCallTarget()));
-                    return false;
-                } else {
-                    return true;
+                } else if (node instanceof GeneratorExpressionNode) {
+                    constants.add(new PCode(PythonBuiltinClassType.PCode, ((GeneratorExpressionNode) node).getCallTarget()));
                 }
+                return true;
             }
         });
         return constants.toArray();