Fix get_locals

palaviv · palaviv · commit da577306926e · 2019-03-04T21:52:20.000+02:00
diff --git a/tests/snippets/builtin_locals.py b/tests/snippets/builtin_locals.py
@@ -0,0 +1,8 @@
+
+a = 5
+b = 6
+
+loc = locals()
+
+assert loc['a'] == 5
+assert loc['b'] == 6
diff --git a/vm/src/vm.rs b/vm/src/vm.rs
@@ -208,16 +208,8 @@ impl VirtualMachine {
     }
 
     pub fn get_locals(&self) -> PyObjectRef {
-        // let scope = &self.frames.last().unwrap().locals;
-        // scope.clone()
-        // TODO: fix this!
-        self.get_none()
-        /*
-        match (*scope).payload {
-            PyObjectPayload::Scope { scope } => { scope.locals.clone() },
-            _ => { panic!("Should be scope") },
-        } // .clone()
-        */
+        let scope = self.current_scope();
+        scope.locals.clone()
     }
 
     pub fn context(&self) -> &PyContext {
