def test_fix_getcwd(): code = """ getcwd() os.getcwd() """ new_code = LegacyPythonTranslator(code).translate() assert new_code.count('getcwd(') == 0 assert new_code.count('getcwdu(') == 2
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')