test_frozen.py 1.2 KB

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