diff --git a/libmodernize/fixes/fix_int_long_tuple.py b/libmodernize/fixes/fix_int_long_tuple.py index eb6b4fa..71302ea 100644 --- a/libmodernize/fixes/fix_int_long_tuple.py +++ b/libmodernize/fixes/fix_int_long_tuple.py @@ -2,6 +2,7 @@ from lib2to3 import fixer_base from lib2to3 import fixer_util +from libmodernize import touch_import class FixIntLongTuple(fixer_base.BaseFix): @@ -20,6 +21,6 @@ def transform(self, node, results): name = results['name'] name.replace(fixer_util.Name('int', prefix=name.prefix)) else: - fixer_util.touch_import(None, 'six', node) + touch_import(None, 'six', node) pair = results['pair'] pair.replace(fixer_util.Name('six.integer_types', prefix=pair.prefix)) diff --git a/tests/test_fix_int_long_tuple.py b/tests/test_fix_int_long_tuple.py index f65d9be..56df88f 100644 --- a/tests/test_fix_int_long_tuple.py +++ b/tests/test_fix_int_long_tuple.py @@ -6,6 +6,7 @@ INT_LONG_ISINSTANCE = ("""\ isinstance(1, (int, long)) """, """\ +from __future__ import absolute_import import six isinstance(1, six.integer_types) """) @@ -13,6 +14,7 @@ LONG_INT_ISINSTANCE = ("""\ isinstance(1, (long, int)) """, """\ +from __future__ import absolute_import import six isinstance(1, six.integer_types) """)