Make test_builtin work when executed directly

brettcannon · brettcannon · commit 2f9f056db132 · 2013-06-15T23:24:11.000-04:00
diff --git a/Lib/test/test_builtin.py b/Lib/test/test_builtin.py
@@ -584,7 +584,10 @@ def __setitem__(self, key, value):
                 raise frozendict_error("frozendict is readonly")
 
         # read-only builtins
-        frozen_builtins = frozendict(__builtins__)
+        if isinstance(__builtins__, types.ModuleType):
+            frozen_builtins = frozendict(__builtins__.__dict__)
+        else:
+            frozen_builtins = frozendict(__builtins__)
         code = compile("__builtins__['superglobal']=2; print(superglobal)", "test", "exec")
         self.assertRaises(frozendict_error,
                           exec, code, {'__builtins__': frozen_builtins})
