Small restructuring.

This commit is contained in:
Dave Halter
2017-03-15 08:56:49 +01:00
parent 55c9fd3227
commit 375749c5c3
2 changed files with 5 additions and 3 deletions

View File

@@ -615,7 +615,9 @@ class ImplicitNamespaceContext(use_metaclass(CachedMetaClass, context.TreeContex
paths = self.paths
file_names = chain.from_iterable(os.listdir(path) for path in paths)
mods = [file_name.rpartition('.')[0] if '.' in file_name else file_name for file_name in file_names]
mods = [
file_name.rpartition('.')[0] if '.' in file_name else file_name
for file_name in file_names]
for name in mods:
names[name] = imports.SubModuleName(self, name)

View File

@@ -29,7 +29,7 @@ def test_implicit_namespace_package():
# completion
completions = script_with_path('from pkg import ').completions()
names = [str(c.name) for c in completions] # str because of unicode
names = [c.name for c in completions]
compare = ['ns1_file', 'ns2_file']
# must at least contain these items, other items are not important
assert set(compare) == set(names)