"""Fixer for reload().
-reload(s) -> imp.reload(s)"""
+reload(s) -> importlib.reload(s)"""
# Local imports
from .. import fixer_base
if (obj.type == self.syms.argument and
obj.children[0].value == '**'):
return # Make no change.
- names = ('imp', 'reload')
+ names = ('importlib', 'reload')
new = ImportAndCall(node, results, names)
- touch_import(None, 'imp', node)
+ touch_import(None, 'importlib', node)
return new
def test(self):
b = """reload(a)"""
- a = """import imp\nimp.reload(a)"""
+ a = """import importlib\nimportlib.reload(a)"""
self.check(b, a)
def test_comment(self):
b = """reload( a ) # comment"""
- a = """import imp\nimp.reload( a ) # comment"""
+ a = """import importlib\nimportlib.reload( a ) # comment"""
self.check(b, a)
# PEP 8 comments
b = """reload( a ) # comment"""
- a = """import imp\nimp.reload( a ) # comment"""
+ a = """import importlib\nimportlib.reload( a ) # comment"""
self.check(b, a)
def test_space(self):
b = """reload( a )"""
- a = """import imp\nimp.reload( a )"""
+ a = """import importlib\nimportlib.reload( a )"""
self.check(b, a)
b = """reload( a)"""
- a = """import imp\nimp.reload( a)"""
+ a = """import importlib\nimportlib.reload( a)"""
self.check(b, a)
b = """reload(a )"""
- a = """import imp\nimp.reload(a )"""
+ a = """import importlib\nimportlib.reload(a )"""
self.check(b, a)
def test_unchanged(self):