summaryrefslogtreecommitdiffstats
path: root/Lib/importlib/test/frozen/test_finder.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/importlib/test/frozen/test_finder.py')
-rw-r--r--Lib/importlib/test/frozen/test_finder.py44
1 files changed, 44 insertions, 0 deletions
diff --git a/Lib/importlib/test/frozen/test_finder.py b/Lib/importlib/test/frozen/test_finder.py
new file mode 100644
index 0000000..2541019
--- /dev/null
+++ b/Lib/importlib/test/frozen/test_finder.py
@@ -0,0 +1,44 @@
+import importlib
+from ..builtin import test_finder
+from .. import support
+
+import unittest
+
+
+class FinderTests(test_finder.FinderTests):
+
+ """Test finding frozen modules."""
+
+ def find(self, name, path=None):
+ finder = importlib.FrozenImporter()
+ return finder.find_module(name, path)
+
+
+ def test_module(self):
+ name = '__hello__'
+ loader = self.find(name)
+ self.assert_(hasattr(loader, 'load_module'))
+
+ def test_package(self):
+ loader = self.find('__phello__')
+ self.assert_(hasattr(loader, 'load_module'))
+
+ def test_module_in_package(self):
+ loader = self.find('__phello__.spam', ['__phello__'])
+ self.assert_(hasattr(loader, 'load_module'))
+
+ def test_package_in_package(self):
+ pass
+
+ def test_failure(self):
+ loader = self.find('<not real>')
+ self.assert_(loader is None)
+
+
+def test_main():
+ from test.support import run_unittest
+ run_unittest(FinderTests)
+
+
+if __name__ == '__main__':
+ test_main()