def test_will_import_basic(): i = ImportRedirector('foobar4', 'foobar4_%s') i.import_func = MagicMock(side_effect=sys_modules_side_effect) mod = i.load_module('foobar4.another') i.import_func.assert_called_with('foobar4_another') assert mod is sys.modules['foobar4.another'] assert mod is sys.modules['foobar4_another'] assert mod is i.module.another
def test_will_import_from_regular_when_builtins_fail(): i = ImportRedirector('foobar6', 'foobar6_%s', builtins='foobar6.builtins') i.import_func = MagicMock(side_effect=sys_modules_side_effect) mod = i.load_module('foobar6.another') i.import_func.assert_called_once_with('foobar6.builtins.another') assert mod is sys.modules['foobar6.another'] assert mod is sys.modules['foobar6.builtins.another'] assert mod is i.module.another
def test_will_return_existing(): i = ImportRedirector('foobar3', 'foobar3_%s') m = i.load_module('sys') assert m is sys