submodules need relative imports not absolute

This commit is contained in:
Dave Halter
2014-06-25 11:08:29 +02:00
parent 44238a9f92
commit 6e82fa31e1
3 changed files with 6 additions and 4 deletions

View File

@@ -187,9 +187,10 @@ class FakeStatement(pr.Statement):
class FakeImport(pr.Import):
def __init__(self, name, parent):
def __init__(self, name, parent, level=0):
p = 0, 0
super(FakeImport, self).__init__(FakeSubModule, p, p, name)
super(FakeImport, self).__init__(FakeSubModule, p, p, name,
relative_count=level)
self.parent = parent

View File

@@ -561,7 +561,8 @@ class ModuleWrapper(pr.Module):
mods = pkgutil.iter_modules([os.path.dirname(path)])
for module_loader, name, is_pkg in mods:
name = helpers.FakeName(name)
imp = helpers.FakeImport(name, self)
# It's obviously a relative import to the current module.
imp = helpers.FakeImport(name, self, level=1)
name.parent = imp
names.append(name)
return names

View File

@@ -76,7 +76,7 @@ return_one(''.undefined_attribute)
[r for r in [1, 2]]
# some random error throw
# some random error that showed up
class NotCalled():
def match_something(self, param):
seems_to_need_an_assignment = param