]> granicus.if.org Git - python/commitdiff
Make test_builtin work when executed directly
authorBrett Cannon <brett@python.org>
Sun, 16 Jun 2013 03:24:11 +0000 (23:24 -0400)
committerBrett Cannon <brett@python.org>
Sun, 16 Jun 2013 03:24:11 +0000 (23:24 -0400)
Lib/test/test_builtin.py

index f46f8d561ca80a2221dbc39886616166dd161d35..f8504564db294c3f7964426a537b7d335409e8d7 100644 (file)
@@ -584,7 +584,10 @@ class BuiltinTest(unittest.TestCase):
                 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})