47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
# Test the frozen module defined in frozen.c.
|
|
from __future__ import with_statement
|
|
|
|
from test.test_support import captured_stdout, run_unittest
|
|
import unittest
|
|
import sys, os
|
|
|
|
class FrozenTests(unittest.TestCase):
|
|
def test_frozen(self):
|
|
try:
|
|
import __hello__
|
|
except ImportError as x:
|
|
self.fail("import __hello__ failed:" + str(x))
|
|
self.assertEqual(__hello__.initialized, True)
|
|
self.assertEqual(len(dir(__hello__)), 6, dir(__hello__))
|
|
|
|
try:
|
|
import __phello__
|
|
except ImportError as x:
|
|
self.fail("import __phello__ failed:" + str(x))
|
|
self.assertEqual(__phello__.initialized, True)
|
|
if not "__phello__.spam" in sys.modules:
|
|
self.assertEqual(len(dir(__phello__)), 7, dir(__phello__))
|
|
else:
|
|
self.assertEqual(len(dir(__phello__)), 8, dir(__phello__))
|
|
|
|
try:
|
|
import __phello__.spam
|
|
except ImportError as x:
|
|
self.fail("import __phello__.spam failed:" + str(x))
|
|
self.assertEqual(__phello__.spam.initialized, True)
|
|
self.assertEqual(len(dir(__phello__.spam)), 6)
|
|
self.assertEqual(len(dir(__phello__)), 8)
|
|
|
|
try:
|
|
import __phello__.foo
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
self.fail("import __phello__.foo should have failed")
|
|
|
|
def test_main():
|
|
run_unittest(FrozenTests)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|