def test_fix_future(): code = """ foo = 2 """ new_code = LegacyPythonTranslator(code).translate() assert new_code.count('from __future__ import ') == 1 assert new_code.index('__future__') < new_code.index('foo') code = """ # bla 'docstring' foo = 2 """ new_code = LegacyPythonTranslator(code).translate() assert new_code.count('from __future__ import ') == 1 assert new_code.index('__future__') < new_code.index('foo') assert new_code.index('__future__') > new_code.index('bla') assert new_code.index('__future__') > new_code.index('docstring')