1# Test the frozen module defined in frozen.c. 2 3from test.test_support import captured_stdout, run_unittest 4import unittest 5import sys 6 7class FrozenTests(unittest.TestCase): 8 def test_frozen(self): 9 10 with captured_stdout() as stdout: 11 try: 12 import __hello__ 13 except ImportError, x: 14 self.fail("import __hello__ failed:" + str(x)) 15 16 try: 17 import __phello__ 18 except ImportError, x: 19 self.fail("import __phello__ failed:" + str(x)) 20 21 try: 22 import __phello__.spam 23 except ImportError, x: 24 self.fail("import __phello__.spam failed:" + str(x)) 25 26 try: 27 import __phello__.foo 28 except ImportError: 29 pass 30 else: 31 self.fail("import __phello__.foo should have failed") 32 33 self.assertEqual(stdout.getvalue(), 34 'Hello world...\nHello world...\nHello world...\n') 35 36 del sys.modules['__hello__'] 37 del sys.modules['__phello__'] 38 del sys.modules['__phello__.spam'] 39 40 41def test_main(): 42 run_unittest(FrozenTests) 43 44 45 46if __name__ == '__main__': 47 test_main() 48