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')
