def run_main(self, *imported):
self.messages = []
code = generate_code(*imported)
code = '#!/usr/bin/python\n# -*- coding: utf-8 -*-\n' + code
tmp_file = tempfile.mkstemp()[1]
try:
with codecs.open(tmp_file, 'w', 'utf-8') as f:
f.write(code)
flake8_future_import.main([tmp_file])
finally:
os.remove(tmp_file)
self.run_test(self.reverse_parse(self.messages), imported)
def run_main(self, *imported):
self.messages = []
code = generate_code(*imported)
code = '#!/usr/bin/python\n# -*- coding: utf-8 -*-\n' + code
tmp_file = tempfile.mkstemp()[1]
try:
with codecs.open(tmp_file, 'w', 'utf-8') as f:
f.write(code)
flake8_future_import.main([tmp_file])
finally:
os.remove(tmp_file)
self.run_test(self.reverse_parse(self.messages), imported)
def run_main(self, *imported):
def iterator():
for fn, line, msg in self.messages:
yield int(line), msg
self.assertEqual(fn, tmp_file)
self.messages = []
code = generate_code(*imported)
code = '#!/usr/bin/python\n# -*- coding: utf-8 -*-\n' + code
tmp_file = tempfile.mkstemp()[1]
try:
with codecs.open(tmp_file, 'w', 'utf-8') as f:
f.write(code)
flake8_future_import.main([tmp_file])
finally:
os.remove(tmp_file)
self.run_test(iterator(), *imported)