Skip to content

Commit 32baf14

Browse files
committed
Use our own touch_import in fix_int_long_tuple, so absolute_imports is added
1 parent d175d40 commit 32baf14

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

libmodernize/fixes/fix_int_long_tuple.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
from lib2to3 import fixer_base
44
from lib2to3 import fixer_util
5+
from libmodernize import touch_import
56

67

78
class FixIntLongTuple(fixer_base.BaseFix):
@@ -20,6 +21,6 @@ def transform(self, node, results):
2021
name = results['name']
2122
name.replace(fixer_util.Name('int', prefix=name.prefix))
2223
else:
23-
fixer_util.touch_import(None, 'six', node)
24+
touch_import(None, 'six', node)
2425
pair = results['pair']
2526
pair.replace(fixer_util.Name('six.integer_types', prefix=pair.prefix))

tests/test_fix_int_long_tuple.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,15 @@
66
INT_LONG_ISINSTANCE = ("""\
77
isinstance(1, (int, long))
88
""", """\
9+
from __future__ import absolute_import
910
import six
1011
isinstance(1, six.integer_types)
1112
""")
1213

1314
LONG_INT_ISINSTANCE = ("""\
1415
isinstance(1, (long, int))
1516
""", """\
17+
from __future__ import absolute_import
1618
import six
1719
isinstance(1, six.integer_types)
1820
""")

0 commit comments

Comments
 (0)