mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-17 14:16:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Merged revisions 65137 via svnmerge from

Browse source 
svn+ssh://pythondev@svn.python.org/sandbox/trunk/2to3/lib2to3

........
  r65137 | georg.brandl | 2008-07-19 08:32:57 -0500 (Sat, 19 Jul 2008) | 2 lines

  #3334: correctly set prefix of imports.
........
This commit is contained in:
Benjamin Peterson 2008-07-19 14:14:06 +00:00
parent b46d6ff279
commit 0147a761b1
2 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
Lib/lib2to3/fixes/fix_import.py
View file

@ -45,7 +45,7 @@ class FixImport(fixer_base.BaseFix):
node.changed() node.changed()
else: else:
new = FromImport('.', getattr(imp, 'content', None) or [imp]) new = FromImport('.', getattr(imp, 'content', None) or [imp])
new.prefix = node.get_prefix() new.set_prefix(node.get_prefix())
node = new node = new
return node return node

11
Lib/lib2to3/tests/test_fixers.py
View file

@ -3319,6 +3319,17 @@ class Test_import(FixerTestCase):
a = "from . import foo.bar as bang" a = "from . import foo.bar as bang"
self.check_both(b, a) self.check_both(b, a)
def test_prefix(self):
b = """
# prefix
import foo.bar
"""
a = """
# prefix
from . import foo.bar
"""
self.check_both(b, a)
if __name__ == "__main__": if __name__ == "__main__":
import __main__ import __main__