Python fix_file_contentsの例

コード例 #1

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

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

コード例 #2

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

コード例 #3

ファイル: main_test.py プロジェクト: bukzor/reorder_python_imports

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

コード例 #4

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

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

コード例 #5

ファイル: main_test.py プロジェクト: bukzor/reorder_python_imports

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

コード例 #6

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

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

コード例 #7

ファイル: main_test.py プロジェクト: bukzor/reorder_python_imports

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

コード例 #8

ファイル: main_test.py プロジェクト: bukzor/reorder_python_imports

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'

コード例 #9

ファイル: main_test.py プロジェクト: bukzor/reorder_python_imports

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'

コード例 #10

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

def test_add_import_trivial():
    assert fix_file_contents(
        '',
        imports_to_add=('from __future__ import absolute_import', ),
    ) == ''

コード例 #11

ファイル: main_test.py プロジェクト: bukzor/reorder_python_imports

def test_remove_import_trivial():
    assert fix_file_contents(
        '',
        imports_to_remove=('from __future__ import with_statement',),
    ) == ''

コード例 #12

ファイル: main_test.py プロジェクト: bukzor/reorder_python_imports

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'

コード例 #13

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

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'

コード例 #14

ファイル: main_test.py プロジェクト: bukzor/reorder_python_imports

def test_add_import_trivial():
    assert fix_file_contents(
        '',
        imports_to_add=('from __future__ import absolute_import',),
    ) == ''

コード例 #15

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

def test_remove_import_trivial():
    assert fix_file_contents(
        '',
        imports_to_remove=('from __future__ import with_statement', ),
    ) == ''

コード例 #16

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

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

コード例 #17

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

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'

コード例 #18

ファイル: main_test.py プロジェクト: rupeshparab/reorder_python_imports

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'