in a generator, the locals() call should give the generator frame locals

Author: timfel

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/BuiltinFunctions.java

@@ -1785,11 +1785,17 @@ public Object globals(VirtualFrame frame) {
     abstract static class LocalsNode extends PythonBuiltinNode {
         @Child ReadCallerFrameNode readCallerFrameNode = ReadCallerFrameNode.create();
         @Child GetLocalsNode getLocalsNode = GetLocalsNode.create();
+        private final ConditionProfile inGenerator = ConditionProfile.createBinaryProfile();
 
         @Specialization
         public Object locals(VirtualFrame frame) {
             Frame callerFrame = readCallerFrameNode.executeWith(frame);
-            return getLocalsNode.execute(callerFrame);
+            Frame generatorFrame = PArguments.getGeneratorFrame(callerFrame);
+            if (inGenerator.profile(generatorFrame == null)) {
+                return getLocalsNode.execute(callerFrame);
+            } else {
+                return getLocalsNode.execute(generatorFrame);
+            }
         }
     }
 }