Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 1 | # Test the frozen module defined in frozen.c. |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 2 | from __future__ import with_statement |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 3 | |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 4 | from test.test_support import captured_stdout, run_unittest |
| 5 | import unittest |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 6 | import sys, os |
Christian Heimes | f19169f | 2007-11-12 19:19:07 +0000 | [diff] [blame^] | 7 | import warnings |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 8 | |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 9 | class FrozenTests(unittest.TestCase): |
| 10 | def test_frozen(self): |
Christian Heimes | f19169f | 2007-11-12 19:19:07 +0000 | [diff] [blame^] | 11 | try: |
| 12 | import __hello__ |
| 13 | except ImportError as x: |
| 14 | self.fail("import __hello__ failed:" + str(x)) |
| 15 | self.assertEqual(__hello__.initialized, True) |
| 16 | self.assertEqual(len(dir(__hello__)), 5) |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 17 | |
Christian Heimes | f19169f | 2007-11-12 19:19:07 +0000 | [diff] [blame^] | 18 | try: |
| 19 | import __phello__ |
| 20 | except ImportError as x: |
| 21 | self.fail("import __phello__ failed:" + str(x)) |
| 22 | self.assertEqual(__phello__.initialized, True) |
| 23 | if not "__phello__.spam" in sys.modules: |
| 24 | self.assertEqual(len(dir(__phello__)), 6, dir(__phello__)) |
| 25 | else: |
| 26 | self.assertEqual(len(dir(__phello__)), 7, dir(__phello__)) |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 27 | |
Christian Heimes | f19169f | 2007-11-12 19:19:07 +0000 | [diff] [blame^] | 28 | try: |
| 29 | import __phello__.spam |
| 30 | except ImportError as x: |
| 31 | self.fail("import __phello__.spam failed:" + str(x)) |
| 32 | self.assertEqual(__phello__.spam.initialized, True) |
| 33 | self.assertEqual(len(dir(__phello__.spam)), 5) |
| 34 | self.assertEqual(len(dir(__phello__)), 7) |
Guido van Rossum | 4114a4a | 2001-10-18 18:49:37 +0000 | [diff] [blame] | 35 | |
Christian Heimes | f19169f | 2007-11-12 19:19:07 +0000 | [diff] [blame^] | 36 | try: |
| 37 | import __phello__.foo |
| 38 | except ImportError: |
| 39 | pass |
| 40 | else: |
| 41 | self.fail("import __phello__.foo should have failed") |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 42 | |
| 43 | def test_main(): |
| 44 | run_unittest(FrozenTests) |
Guido van Rossum | 3b7210d | 2007-10-15 00:25:56 +0000 | [diff] [blame] | 45 | |
| 46 | if __name__ == "__main__": |
| 47 | test_main() |