| # Test the frozen module defined in frozen.c. |
| |
| from test.test_support import captured_stdout, run_unittest |
| import unittest |
| import sys |
| |
| class FrozenTests(unittest.TestCase): |
| def test_frozen(self): |
| |
| with captured_stdout() as stdout: |
| try: |
| import __hello__ |
| except ImportError, x: |
| self.fail("import __hello__ failed:" + str(x)) |
| |
| try: |
| import __phello__ |
| except ImportError, x: |
| self.fail("import __phello__ failed:" + str(x)) |
| |
| try: |
| import __phello__.spam |
| except ImportError, x: |
| self.fail("import __phello__.spam failed:" + str(x)) |
| |
| try: |
| import __phello__.foo |
| except ImportError: |
| pass |
| else: |
| self.fail("import __phello__.foo should have failed") |
| |
| self.assertEqual(stdout.getvalue(), |
| 'Hello world...\nHello world...\nHello world...\n') |
| |
| del sys.modules['__hello__'] |
| del sys.modules['__phello__'] |
| del sys.modules['__phello__.spam'] |
| |
| |
| def test_main(): |
| run_unittest(FrozenTests) |
| |
| |
| |
| if __name__ == '__main__': |
| test_main() |