def test_from_import_object(self):
# from foo.foo_c import obj_c
modset = ModuleSet([FOO.init, FOO.a, FOO.b, FOO.c])
got = modset.get_imports(modset.by_name['foo.foo_a'])
assert len(got) == 2
assert FOO.b in got # doesnt matter for this test
assert FOO.c in got
def test_init_no_packge(self):
# if a module of a package is added but no __init__.py
# its packages is not added to the list of packages
modset = ModuleSet([FOO.a])
assert 0 == len(modset.pkgs)
assert 1 == len(modset.by_path)
assert modset.by_path[FOO.a].fqn == ['foo', 'foo_a']
def test_init_with_packge(self):
modset = ModuleSet([FOO.init, FOO.a])
assert set(['foo']) == modset.pkgs
assert 2 == len(modset.by_path)
assert modset.by_path[FOO.init].fqn == ['foo', '__init__']
assert modset.by_path[FOO.a].fqn == ['foo', 'foo_a']
assert 2 == len(modset.by_name)
assert modset.by_name['foo.__init__'].fqn == ['foo', '__init__']
assert modset.by_name['foo.foo_a'].fqn == ['foo', 'foo_a']
def test_relative_intra_pkg_obj(self):
# from sub.sub_A import obj_sub_a_xxx
modset = ModuleSet([FOO.init, SUB.init, SUB.a, SUB.b])
got = modset.get_imports(modset.by_name['foo.sub.sub_b'])
assert len(got) == 1
assert SUB.a in got
def test_relative_sub_import_mod(self):
# from sub import sub_a
modset = ModuleSet([FOO.init, FOO.f, SUB.init, SUB.a])
got = modset.get_imports(modset.by_name['foo.foo_f'])
assert len(got) == 1
assert SUB.a in got
def test_relative_sub_import_obj(self):
# from sub.sub_a import obj_sub_a_xxx
modset = ModuleSet([FOO.init, FOO.e, SUB.init, SUB.a])
got = modset.get_imports(modset.by_name['foo.foo_e'])
assert len(got) == 1
assert SUB.a in got
def test_relative_parent(self):
# from .. import foo_d
modset = ModuleSet([FOO.init, FOO.d, SUB.init, SUB.a])
got = modset.get_imports(modset.by_name['foo.sub.sub_a'])
assert len(got) == 1
assert FOO.d in got
def test_import_pkg(self):
# import foo
modset = ModuleSet([FOO.init, BAR])
got = modset.get_imports(modset.by_name['bar'])
assert len(got) == 1
assert FOO.init in got
def test_relative_intra_import_pkg_obj(self):
# from . import foo_i
modset = ModuleSet([FOO.init, FOO.c])
got = modset.get_imports(modset.by_name['foo.foo_c'])
assert len(got) == 1
assert FOO.init in got
def test_relative_import_old(self):
# import foo_c
modset = ModuleSet([FOO.init, FOO.b, FOO.c])
got = modset.get_imports(modset.by_name['foo.foo_b'])
assert len(got) == 1
assert FOO.c in got
def test_import_obj(self):
# import baz.obj_baz
modset = ModuleSet([FOO.b, BAZ])
got = modset.get_imports(modset.by_name['foo.foo_b'])
assert len(got) == 1
assert BAZ in got
def test_from_pkg_import_obj(self):
# from foo import obj_1
modset = ModuleSet([FOO.init, BAZ])
got = modset.get_imports(modset.by_name['baz'])
assert len(got) == 1
assert FOO.init in got
def test_from_pkg_import_module(self):
# from foo import foo_b
modset = ModuleSet([FOO.init, FOO.a, FOO.b])
got = modset.get_imports(modset.by_name['foo.foo_a'])
assert len(got) == 1
assert FOO.b in got
def test_init_subpackge(self):
modset = ModuleSet([FOO.init, SUB.init, SUB.a])
assert set(['foo', 'foo.sub']) == modset.pkgs
assert 3 == len(modset.by_path)
assert modset.by_path[SUB.a].fqn == ['foo', 'sub', 'sub_a']
def test_relative_intra_import_module(self):
# from . import foo_c
modset = ModuleSet([FOO.init, FOO.c, FOO.d])
got = modset.get_imports(modset.by_name['foo.foo_d'])
assert len(got) == 1
assert FOO.c in got
def test_import_not_tracked(self):
modset = ModuleSet([FOO.a])
got = modset.get_imports(modset.by_name['foo.foo_a'])
assert len(got) == 0
def test_import_module(self):
# import bar
modset = ModuleSet([FOO.a, BAR])
got = modset.get_imports(modset.by_name['foo.foo_a'])
assert len(got) == 1
assert BAR in got
