def test_import_package_modules_from_init(project, tmpdir): project.set_root(str(tmpdir)) pkgdir = tmpdir.join('package') pkgdir.mkdir() source, pos = get_source_and_pos(''' import module def foo(): module.n| ''') pkg = pkgdir.join('__init__.py') pkg.write(source) m = pkgdir.join('module.py') m.write('name = []') match, result = assist(project, source, pos, str(pkg)) assert result == ['name']
def assist(self, path, source, position, filename): return assist(self.get_project(path), source, position, filename)
def do_assist(project, source, filename=None): filename = filename or 'test.py' source, pos = get_source_and_pos(source) return assist(project, source, pos, filename)[1]