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)
