| from importlib import util |
| from . import util as test_util |
| import imp |
| import sys |
| import types |
| import unittest |
| |
| |
| class ModuleForLoaderTests(unittest.TestCase): |
| |
| """Tests for importlib.util.module_for_loader.""" |
| |
| def return_module(self, name): |
| fxn = util.module_for_loader(lambda self, module: module) |
| return fxn(self, name) |
| |
| def raise_exception(self, name): |
| def to_wrap(self, module): |
| raise ImportError |
| fxn = util.module_for_loader(to_wrap) |
| try: |
| fxn(self, name) |
| except ImportError: |
| pass |
| |
| def test_new_module(self): |
| # Test that when no module exists in sys.modules a new module is |
| # created. |
| module_name = 'a.b.c' |
| with test_util.uncache(module_name): |
| module = self.return_module(module_name) |
| self.assertTrue(module_name in sys.modules) |
| self.assertTrue(isinstance(module, types.ModuleType)) |
| self.assertEqual(module.__name__, module_name) |
| |
| def test_reload(self): |
| # Test that a module is reused if already in sys.modules. |
| name = 'a.b.c' |
| module = imp.new_module('a.b.c') |
| with test_util.uncache(name): |
| sys.modules[name] = module |
| returned_module = self.return_module(name) |
| self.assertIs(returned_module, sys.modules[name]) |
| |
| def test_new_module_failure(self): |
| # Test that a module is removed from sys.modules if added but an |
| # exception is raised. |
| name = 'a.b.c' |
| with test_util.uncache(name): |
| self.raise_exception(name) |
| self.assertTrue(name not in sys.modules) |
| |
| def test_reload_failure(self): |
| # Test that a failure on reload leaves the module in-place. |
| name = 'a.b.c' |
| module = imp.new_module(name) |
| with test_util.uncache(name): |
| sys.modules[name] = module |
| self.raise_exception(name) |
| self.assertIs(module, sys.modules[name]) |
| |
| |
| class SetPackageTests(unittest.TestCase): |
| |
| |
| """Tests for importlib.util.set_package.""" |
| |
| def verify(self, module, expect): |
| """Verify the module has the expected value for __package__ after |
| passing through set_package.""" |
| fxn = lambda: module |
| wrapped = util.set_package(fxn) |
| wrapped() |
| self.assertTrue(hasattr(module, '__package__')) |
| self.assertEqual(expect, module.__package__) |
| |
| def test_top_level(self): |
| # __package__ should be set to the empty string if a top-level module. |
| # Implicitly tests when package is set to None. |
| module = imp.new_module('module') |
| module.__package__ = None |
| self.verify(module, '') |
| |
| def test_package(self): |
| # Test setting __package__ for a package. |
| module = imp.new_module('pkg') |
| module.__path__ = ['<path>'] |
| module.__package__ = None |
| self.verify(module, 'pkg') |
| |
| def test_submodule(self): |
| # Test __package__ for a module in a package. |
| module = imp.new_module('pkg.mod') |
| module.__package__ = None |
| self.verify(module, 'pkg') |
| |
| def test_setting_if_missing(self): |
| # __package__ should be set if it is missing. |
| module = imp.new_module('mod') |
| if hasattr(module, '__package__'): |
| delattr(module, '__package__') |
| self.verify(module, '') |
| |
| def test_leaving_alone(self): |
| # If __package__ is set and not None then leave it alone. |
| for value in (True, False): |
| module = imp.new_module('mod') |
| module.__package__ = value |
| self.verify(module, value) |
| |
| |
| def test_main(): |
| from test import support |
| support.run_unittest(ModuleForLoaderTests, SetPackageTests) |
| |
| |
| if __name__ == '__main__': |
| test_main() |