| 1234567891011121314151617181920212223242526272829303132333435363738 |
- import sys
- import importlib
- from importlib.util import LazyLoader, find_spec, module_from_spec
- import pytest
- # Warning raised by _reload_guard() in numpy/__init__.py
- @pytest.mark.filterwarnings("ignore:The NumPy module was reloaded")
- def test_lazy_load():
- # gh-22045. lazyload doesn't import submodule names into the namespace
- # muck with sys.modules to test the importing system
- old_numpy = sys.modules.pop("numpy")
- numpy_modules = {}
- for mod_name, mod in list(sys.modules.items()):
- if mod_name[:6] == "numpy.":
- numpy_modules[mod_name] = mod
- sys.modules.pop(mod_name)
- try:
- # create lazy load of numpy as np
- spec = find_spec("numpy")
- module = module_from_spec(spec)
- sys.modules["numpy"] = module
- loader = LazyLoader(spec.loader)
- loader.exec_module(module)
- np = module
- # test a subpackage import
- from numpy.lib import recfunctions
- # test triggering the import of the package
- np.ndarray
- finally:
- if old_numpy:
- sys.modules["numpy"] = old_numpy
- sys.modules.update(numpy_modules)
|
