| from importlib import _bootstrap |
| from . import util as ext_util |
| from .. import abc |
| from .. import util |
| |
| import sys |
| import unittest |
| |
| |
| class LoaderTests(abc.LoaderTests): |
| |
| """Test load_module() for extension modules.""" |
| |
| def load_module(self, fullname): |
| loader = _bootstrap._ExtensionFileLoader(ext_util.NAME, |
| ext_util.FILEPATH) |
| return loader.load_module(fullname) |
| |
| def test_module(self): |
| with util.uncache(ext_util.NAME): |
| module = self.load_module(ext_util.NAME) |
| for attr, value in [('__name__', ext_util.NAME), |
| ('__file__', ext_util.FILEPATH), |
| ('__package__', '')]: |
| self.assertEqual(getattr(module, attr), value) |
| self.assertTrue(ext_util.NAME in sys.modules) |
| self.assertTrue(isinstance(module.__loader__, |
| _bootstrap._ExtensionFileLoader)) |
| |
| def test_package(self): |
| # Extensions are not found in packages. |
| pass |
| |
| def test_lacking_parent(self): |
| # Extensions are not found in packages. |
| pass |
| |
| def test_module_reuse(self): |
| with util.uncache(ext_util.NAME): |
| module1 = self.load_module(ext_util.NAME) |
| module2 = self.load_module(ext_util.NAME) |
| self.assertTrue(module1 is module2) |
| |
| def test_state_after_failure(self): |
| # No easy way to trigger a failure after a successful import. |
| pass |
| |
| def test_unloadable(self): |
| with self.assertRaises(ImportError): |
| self.load_module('asdfjkl;') |
| |
| |
| def test_main(): |
| from test.support import run_unittest |
| run_unittest(LoaderTests) |
| |
| |
| if __name__ == '__main__': |
| test_main() |