# XXX: overhead to the fixer.
MAPPING = {"StringIO": ("io", ["StringIO"]),
"cStringIO": ("io", ["StringIO"]),
+ "cPickle": ("pickle", ['BadPickleGet', 'HIGHEST_PROTOCOL',
+ 'PickleError', 'Pickler', 'PicklingError',
+ 'UnpickleableError', 'Unpickler', 'UnpicklingError',
+ 'compatible_formats', 'dump', 'dumps', 'format_version',
+ 'load', 'loads']),
"__builtin__" : ("builtins", builtin_names),
'copy_reg': ('copyreg', ['pickle',
'constructor',
s = "foo(xreadlines)"
self.unchanged(s)
-# Disable test, as it takes a too long time to run, and also
-# fails in 2.6.
-#class Test_imports(FixerTestCase):
-class Test_imports:
+class Test_imports(FixerTestCase):
fixer = "imports"
from ..fixes.fix_imports import MAPPING as modules