diff options
Diffstat (limited to 'Lib/importlib/test/frozen/test_finder.py')
-rw-r--r-- | Lib/importlib/test/frozen/test_finder.py | 44 |
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() |