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']
Beispiel #2
0
 def assist(self, path, source, position, filename):
     return assist(self.get_project(path), source, position, filename)
Beispiel #3
0
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]