Beta Shell v2.0

File Manager

~ /usr/local/python-3.9/lib/python3.9/test/test_importlib/import_

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	142 B	-rw-r--r--	2022-02-22 16:24:07	Edit Del
__main__.py	58 B	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test___loader__.py	1.82 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test___package__.py	5.51 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test_api.py	3.7 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test_caching.py	3.51 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test_fromlist.py	7.35 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test_meta_path.py	4.22 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test_packages.py	4.44 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test_path.py	10.3 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del
test_relative_imports.py	9.24 KB	-rw-r--r--	2022-02-22 16:24:07	Edit Del

Editing: test_caching.py (3.51 KB)

"""Test that sys.modules is used properly by import."""
from .. import util
import sys
from types import MethodType
import unittest


class UseCache:

    """When it comes to sys.modules, import prefers it over anything else.

    Once a name has been resolved, sys.modules is checked to see if it contains
    the module desired. If so, then it is returned [use cache]. If it is not
    found, then the proper steps are taken to perform the import, but
    sys.modules is still used to return the imported module (e.g., not what a
    loader returns) [from cache on return]. This also applies to imports of
    things contained within a package and thus get assigned as an attribute
    [from cache to attribute] or pulled in thanks to a fromlist import
    [from cache for fromlist]. But if sys.modules contains None then
    ImportError is raised [None in cache].

    """

    def test_using_cache(self):
        # [use cache]
        module_to_use = "some module found!"
        with util.uncache('some_module'):
            sys.modules['some_module'] = module_to_use
            module = self.__import__('some_module')
            self.assertEqual(id(module_to_use), id(module))

    def test_None_in_cache(self):
        #[None in cache]
        name = 'using_None'
        with util.uncache(name):
            sys.modules[name] = None
            with self.assertRaises(ImportError) as cm:
                self.__import__(name)
            self.assertEqual(cm.exception.name, name)


(Frozen_UseCache,
 Source_UseCache
 ) = util.test_both(UseCache, __import__=util.__import__)


class ImportlibUseCache(UseCache, unittest.TestCase):

    # Pertinent only to PEP 302; exec_module() doesn't return a module.

    __import__ = util.__import__['Source']

    def create_mock(self, *names, return_=None):
        mock = util.mock_modules(*names)
        original_load = mock.load_module
        def load_module(self, fullname):
            original_load(fullname)
            return return_
        mock.load_module = MethodType(load_module, mock)
        return mock

    # __import__ inconsistent between loaders and built-in import when it comes
    #   to when to use the module in sys.modules and when not to.
    def test_using_cache_after_loader(self):
        # [from cache on return]
        with self.create_mock('module') as mock:
            with util.import_state(meta_path=[mock]):
                module = self.__import__('module')
                self.assertEqual(id(module), id(sys.modules['module']))

    # See test_using_cache_after_loader() for reasoning.
    def test_using_cache_for_assigning_to_attribute(self):
        # [from cache to attribute]
        with self.create_mock('pkg.__init__', 'pkg.module') as importer:
            with util.import_state(meta_path=[importer]):
                module = self.__import__('pkg.module')
                self.assertTrue(hasattr(module, 'module'))
                self.assertEqual(id(module.module),
                                 id(sys.modules['pkg.module']))

    # See test_using_cache_after_loader() for reasoning.
    def test_using_cache_for_fromlist(self):
        # [from cache for fromlist]
        with self.create_mock('pkg.__init__', 'pkg.module') as importer:
            with util.import_state(meta_path=[importer]):
                module = self.__import__('pkg', fromlist=['module'])
                self.assertTrue(hasattr(module, 'module'))
                self.assertEqual(id(module.module),
                                 id(sys.modules['pkg.module']))


if __name__ == '__main__':
    unittest.main()