added __name__ to scope

rmliddle · rmliddle · commit e7942264d607 · 2018-10-10T22:12:30.000+11:00
diff --git a/tests/snippets/import_name.py b/tests/snippets/import_name.py
@@ -0,0 +1,2 @@
+def import_func():
+    assert __name__ == "import_name"
diff --git a/tests/snippets/name.py b/tests/snippets/name.py
@@ -0,0 +1,9 @@
+#when name.py is run __name__ should equal to __main__
+assert __name__ == "__main__"
+
+from import_name import import_func
+
+#__name__ should be set to import_func
+import_func()
+
+assert __name__ == "__main__"
diff --git a/vm/src/builtins.rs b/vm/src/builtins.rs
@@ -339,7 +339,7 @@ pub fn builtin_print(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
             Ok(v) => objstr::get_value(&v),
             Err(err) => return Err(err),
         };
-        print!("{} ", s);
+        print!("{}", s);
     }
     println!();
     io::stdout().flush().unwrap();
@@ -395,6 +395,11 @@ fn builtin_setattr(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
 pub fn make_module(ctx: &PyContext) -> PyObjectRef {
     // scope[String::from("print")] = print;
     let mut dict = HashMap::new();
+    //set __name__ fixes: https://github.com/RustPython/RustPython/issues/146
+    dict.insert(
+        String::from("__name__"),
+        ctx.new_str(String::from("__main__")),
+    );
     dict.insert(String::from("abs"), ctx.new_rustfunc(builtin_abs));
     dict.insert(String::from("all"), ctx.new_rustfunc(builtin_all));
     dict.insert(String::from("any"), ctx.new_rustfunc(builtin_any));
