def test_rewriting_file(self): import os out_file = 'rerite_me_testOUTPUT.py' try: os.remove(out_file) except OSError: pass rewriter.rewrite('rewrite_me_test.py', condition="GREEN", replace=False, out=out_file) with open(out_file, 'r') as fin: text = fin.read() expected_text = """\ import re def main(): x = "a letter" x = x + " some more letters" print re.match(r'thing', x) print "I'm Green" if __name__ == '__main__': main()""" self.assertEqual(text, expected_text) os.remove(out_file)
def maybe_rewrite(): if iterations() >= max_iterations: cndn = max_conversion([(choice, conversion_rate(choice)) for choice in experiment_options]) for file in rewrite_files: rewriter.rewrite(file, condition=cndn, replace=True)