Add wrapper for testing fixing functionality

This commit is contained in:
Keyan Pishdadian 2015-02-11 14:41:48 -05:00
parent c90c4e5e8c
commit 4a9e34a574

View file

@ -83,8 +83,11 @@ def fix_standard_imports(red):
return red
if __name__ == "__main__":
input_file = sys.argv[1]
def fix(input_file):
ast = read_source(input_file)
new_ast = fix_imports(ast)
write_source(new_ast, input_file)
if __name__ == "__main__":
input_file = sys.argv[1]
fix(input_file)