def test_add_import_not_there():
assert fix_file_contents(
'import os',
imports_to_add=('from __future__ import absolute_import', ),
) == ('from __future__ import absolute_import\n'
'\n'
'import os\n')
def test_does_not_put_before_leading_comment():
assert fix_file_contents(
'# -*- coding: UTF-8 -*-',
imports_to_add=('from __future__ import absolute_import',),
) == (
'# -*- coding: UTF-8 -*-\n'
'from __future__ import absolute_import\n'
)
def test_does_not_put_before_leading_comment():
assert fix_file_contents(
'# -*- coding: UTF-8 -*-',
imports_to_add=('from __future__ import absolute_import',),
) == (
'# -*- coding: UTF-8 -*-\n'
'from __future__ import absolute_import\n'
)
def test_separate_relative_and_separate_from():
ret = fix_file_contents(
'import thirdparty\n'
'from . import bar\n',
separate_from_import=True,
separate_relative=True,
)
assert ret == ('import thirdparty\n' '\n' 'from . import bar\n')
def test_add_import_not_there():
assert fix_file_contents(
'import os',
imports_to_add=('from __future__ import absolute_import',),
) == (
'from __future__ import absolute_import\n'
'\n'
'import os\n'
)
def test_separate_relative_and_separate_from():
ret = fix_file_contents(
'import thirdparty\n'
'from . import bar\n',
separate_from_import=True,
separate_relative=True,
)
assert ret == (
'import thirdparty\n'
'\n'
'from . import bar\n'
)
def test_fix_file_contents(filename):
input_contents = io.open(os.path.join('test_data/inputs', filename)).read()
expected = io.open(os.path.join('test_data/outputs', filename)).read()
assert fix_file_contents(input_contents) == expected
def test_remove_import_import_not_there():
assert fix_file_contents(
'import os\n',
imports_to_remove=('from __future__ import with_statement',),
) == 'import os\n'
def test_remove_imports_actually_removes():
assert fix_file_contents(
'from __future__ import with_statement\n\n'
'import os\n',
imports_to_remove=('from __future__ import with_statement',),
) == 'import os\n'
def test_add_import_trivial():
assert fix_file_contents(
'',
imports_to_add=('from __future__ import absolute_import', ),
) == ''
def test_remove_import_trivial():
assert fix_file_contents(
'',
imports_to_remove=('from __future__ import with_statement',),
) == ''
def test_add_import_import_already_there():
assert fix_file_contents(
'from __future__ import absolute_import\n',
imports_to_add=('from __future__ import absolute_import',),
) == 'from __future__ import absolute_import\n'
def test_add_import_import_already_there():
assert fix_file_contents(
'from __future__ import absolute_import\n',
imports_to_add=('from __future__ import absolute_import', ),
) == 'from __future__ import absolute_import\n'
def test_add_import_trivial():
assert fix_file_contents(
'',
imports_to_add=('from __future__ import absolute_import',),
) == ''
def test_remove_import_trivial():
assert fix_file_contents(
'',
imports_to_remove=('from __future__ import with_statement', ),
) == ''
def test_fix_file_contents(filename):
input_contents = io.open(os.path.join('test_data/inputs', filename)).read()
expected = io.open(os.path.join('test_data/outputs', filename)).read()
assert fix_file_contents(input_contents) == expected
def test_remove_imports_actually_removes():
assert fix_file_contents(
'from __future__ import with_statement\n\n'
'import os\n',
imports_to_remove=('from __future__ import with_statement', ),
) == 'import os\n'
def test_remove_import_import_not_there():
assert fix_file_contents(
'import os\n',
imports_to_remove=('from __future__ import with_statement', ),
) == 'import os\n'