mirror of
https://github.com/python/cpython.git
synced 2024-12-15 21:04:00 +08:00
94 lines
3.6 KiB
Python
94 lines
3.6 KiB
Python
from . import util
|
|
import imp
|
|
import importlib
|
|
import sys
|
|
import unittest
|
|
|
|
|
|
class ImportModuleTests(unittest.TestCase):
|
|
|
|
"""Test importlib.import_module."""
|
|
|
|
def test_module_import(self):
|
|
# Test importing a top-level module.
|
|
with util.mock_modules('top_level') as mock:
|
|
with util.import_state(meta_path=[mock]):
|
|
module = importlib.import_module('top_level')
|
|
self.assertEqual(module.__name__, 'top_level')
|
|
|
|
def test_absolute_package_import(self):
|
|
# Test importing a module from a package with an absolute name.
|
|
pkg_name = 'pkg'
|
|
pkg_long_name = '{0}.__init__'.format(pkg_name)
|
|
name = '{0}.mod'.format(pkg_name)
|
|
with util.mock_modules(pkg_long_name, name) as mock:
|
|
with util.import_state(meta_path=[mock]):
|
|
module = importlib.import_module(name)
|
|
self.assertEqual(module.__name__, name)
|
|
|
|
def test_shallow_relative_package_import(self):
|
|
# Test importing a module from a package through a relative import.
|
|
pkg_name = 'pkg'
|
|
pkg_long_name = '{0}.__init__'.format(pkg_name)
|
|
module_name = 'mod'
|
|
absolute_name = '{0}.{1}'.format(pkg_name, module_name)
|
|
relative_name = '.{0}'.format(module_name)
|
|
with util.mock_modules(pkg_long_name, absolute_name) as mock:
|
|
with util.import_state(meta_path=[mock]):
|
|
importlib.import_module(pkg_name)
|
|
module = importlib.import_module(relative_name, pkg_name)
|
|
self.assertEqual(module.__name__, absolute_name)
|
|
|
|
def test_deep_relative_package_import(self):
|
|
modules = ['a.__init__', 'a.b.__init__', 'a.c']
|
|
with util.mock_modules(*modules) as mock:
|
|
with util.import_state(meta_path=[mock]):
|
|
importlib.import_module('a')
|
|
importlib.import_module('a.b')
|
|
module = importlib.import_module('..c', 'a.b')
|
|
self.assertEqual(module.__name__, 'a.c')
|
|
|
|
def test_absolute_import_with_package(self):
|
|
# Test importing a module from a package with an absolute name with
|
|
# the 'package' argument given.
|
|
pkg_name = 'pkg'
|
|
pkg_long_name = '{0}.__init__'.format(pkg_name)
|
|
name = '{0}.mod'.format(pkg_name)
|
|
with util.mock_modules(pkg_long_name, name) as mock:
|
|
with util.import_state(meta_path=[mock]):
|
|
importlib.import_module(pkg_name)
|
|
module = importlib.import_module(name, pkg_name)
|
|
self.assertEqual(module.__name__, name)
|
|
|
|
def test_relative_import_wo_package(self):
|
|
# Relative imports cannot happen without the 'package' argument being
|
|
# set.
|
|
with self.assertRaises(TypeError):
|
|
importlib.import_module('.support')
|
|
|
|
|
|
def test_loaded_once(self):
|
|
# Issue #13591: Modules should only be loaded once when
|
|
# initializing the parent package attempts to import the
|
|
# module currently being imported.
|
|
b_load_count = 0
|
|
def load_a():
|
|
importlib.import_module('a.b')
|
|
def load_b():
|
|
nonlocal b_load_count
|
|
b_load_count += 1
|
|
code = {'a': load_a, 'a.b': load_b}
|
|
modules = ['a.__init__', 'a.b']
|
|
with util.mock_modules(*modules, module_code=code) as mock:
|
|
with util.import_state(meta_path=[mock]):
|
|
importlib.import_module('a.b')
|
|
self.assertEqual(b_load_count, 1)
|
|
|
|
def test_main():
|
|
from test.support import run_unittest
|
|
run_unittest(ImportModuleTests)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test_main()
|