import statements completion; little bug noticed in #27

This commit is contained in:
David Halter
2012-10-18 00:16:37 +02:00
parent 75c4ed30ee
commit fa9772c82e
2 changed files with 6 additions and 0 deletions

View File

@@ -234,6 +234,10 @@ class Script(object):
# do the magic functions first? and then recheck here?
if not isinstance(s, evaluate.Function):
if isinstance(s, imports.ImportPath):
if like == 'import':
l = self.module.get_line(self.pos[0])[:self.pos[1]]
if not l.endswith('import import'):
continue
names = s.get_defined_names(on_import_stmt=True)
else:
names = s.get_defined_names()

View File

@@ -105,6 +105,8 @@ class TestRegression(unittest.TestCase):
assert 10 < len(self.complete("from .", (1, 5))) < 30
assert 10 < len(self.complete("from . import", (1, 5))) < 30
assert 10 < len(self.complete("from . import classes", (1, 5))) < 30
assert len(self.complete("import")) == 0
assert len(self.complete("import import")) > 0
def test_get_in_function_call(self):
def check(call_def, name, index):