Python split_file Examples

Example #1

def test_split_file():
    input_file = "tests/data/mock_file.txt"
    output_file = "tests/data/mock_output_file.txt"

    splits = split_file(input_file, output_file)
    assert splits == [input_file]

    expected = ["tests/data/mock_output_file.txt.0", "tests/data/mock_output_file.txt.1"]
    splits = split_file(input_file, output_file, 2)
    assert splits == expected
    assert compare_file_contents(input_file, expected)
    cleanup(splits)

    expected = [
        "tests/data/mock_output_file.txt.0",
        "tests/data/mock_output_file.txt.1",
        "tests/data/mock_output_file.txt.2",
    ]
    splits = split_file(input_file, output_file, 3)
    assert splits == expected
    assert compare_file_contents(input_file, expected)
    cleanup(splits)

    expected = [
        "tests/data/mock_output_file.txt.0",
        "tests/data/mock_output_file.txt.1",
        "tests/data/mock_output_file.txt.2",
        "tests/data/mock_output_file.txt.3",
        "tests/data/mock_output_file.txt.4",
    ]
    splits = split_file(input_file, output_file, 5)
    assert splits == expected
    assert compare_file_contents(input_file, expected)
    cleanup(splits)

Example #2

File: test_utility.py Project: priya-gitTest/Data-Load-and-Copy-using-Python

def test_split_file():
    input_file = 'tests/data/mock_file.txt'
    output_file = 'tests/data/mock_output_file.txt'

    expected = ['tests/data/mock_output_file.txt.0',
                'tests/data/mock_output_file.txt.1']
    splits = split_file(input_file, output_file)
    assert splits == expected
    assert compare_file_contents(input_file, expected)
    cleanup(splits)


    expected = ['tests/data/mock_output_file.txt.0',
                'tests/data/mock_output_file.txt.1',
                'tests/data/mock_output_file.txt.2']
    splits = split_file(input_file, output_file, 3)
    assert splits == expected
    assert compare_file_contents(input_file, expected)
    cleanup(splits)


    expected = ['tests/data/mock_output_file.txt.0',
                'tests/data/mock_output_file.txt.1',
                'tests/data/mock_output_file.txt.2',
                'tests/data/mock_output_file.txt.3',
                'tests/data/mock_output_file.txt.4']
    splits = split_file(input_file, output_file, 5)
    assert splits == expected
    assert compare_file_contents(input_file, expected)
    cleanup(splits)

Example #3

File: test_utility.py Project: isabella232/Data-Load-and-Copy-using-Python

def test_split_file_header():
    input_file = "tests/data/mock_file_header.txt"
    input_file_no_header = "tests/data/mock_file.txt"
    output_file = "tests/data/mock_output_file_header.txt"

    splits = split_file(input_file, output_file, ignore_header=1)
    assert splits == [input_file]

    expected = [
        "tests/data/mock_output_file_header.txt.0",
        "tests/data/mock_output_file_header.txt.1",
    ]
    splits = split_file(input_file, output_file, 2, ignore_header=1)
    assert splits == expected
    assert compare_file_contents(input_file_no_header, expected)
    cleanup(splits)

    expected = [
        "tests/data/mock_output_file_header.txt.0",
        "tests/data/mock_output_file_header.txt.1",
        "tests/data/mock_output_file_header.txt.2",
    ]
    splits = split_file(input_file, output_file, 3, ignore_header=1)
    assert splits == expected
    assert compare_file_contents(input_file_no_header, expected)
    cleanup(splits)

    expected = [
        "tests/data/mock_output_file_header.txt.0",
        "tests/data/mock_output_file_header.txt.1",
        "tests/data/mock_output_file_header.txt.2",
        "tests/data/mock_output_file_header.txt.3",
        "tests/data/mock_output_file_header.txt.4",
    ]
    splits = split_file(input_file, output_file, 5, ignore_header=1)
    assert splits == expected
    assert compare_file_contents(input_file_no_header, expected)
    cleanup(splits)

Example #4

File: test_utility.py Project: isabella232/Data-Load-and-Copy-using-Python

def test_split_file_exception():
    input_file = "tests/data/mock_file.txt"
    output_file = "tests/data/mock_output_file.txt"

    if sys.version_info.major == 3:
        builtin_module_name = "builtins"
    else:
        builtin_module_name = "__builtin__"

    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, -1)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, 0)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, 5.65)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, "123")
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, "Test")

    with mock.patch("{0}.next".format(builtin_module_name)) as mock_next:
        with mock.patch("os.remove") as mock_remove:
            mock_next.side_effect = Exception("SomeException")

            with pytest.raises(LocopySplitError):
                split_file(input_file, output_file, 2)
            assert mock_remove.call_count == 2
            mock_remove.reset_mock()

            with pytest.raises(LocopySplitError):
                split_file(input_file, output_file, 3)
            assert mock_remove.call_count == 3

    cleanup([
        "tests/data/mock_output_file.txt.0",
        "tests/data/mock_output_file.txt.1",
        "tests/data/mock_output_file.txt.2",
    ])

Example #5

File: test_utility.py Project: priya-gitTest/Data-Load-and-Copy-using-Python

def test_split_file_exception():
    input_file = 'tests/data/mock_file.txt'
    output_file = 'tests/data/mock_output_file.txt'

    if sys.version_info.major == 3:
        builtin_module_name = 'builtins'
    else:
        builtin_module_name = '__builtin__'

    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, -1)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, 0)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, 1)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, 5.65)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, "123")
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, "Test")


    with mock.patch('{0}.next'.format(builtin_module_name)) as mock_next:
        with mock.patch('os.remove') as mock_remove:
            mock_next.side_effect = Exception('SomeException')

            with pytest.raises(LocopySplitError):
                split_file(input_file, output_file)
            assert mock_remove.call_count == 2
            mock_remove.reset_mock()

            with pytest.raises(LocopySplitError):
                split_file(input_file, output_file, 3)
            assert mock_remove.call_count == 3

    cleanup(['tests/data/mock_output_file.txt.0',
             'tests/data/mock_output_file.txt.1',
             'tests/data/mock_output_file.txt.2'])

Example #6

File: test_utility.py Project: paschmaria/locopy

def test_split_file_exception():
    input_file = "tests/data/mock_file.txt"
    output_file = "tests/data/mock_output_file.txt"

    if sys.version_info.major == 3:
        builtin_module_name = "builtins"
    else:
        builtin_module_name = "__builtin__"

    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, -1)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, 0)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, 5.65)
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, "123")
    with pytest.raises(LocopySplitError):
        split_file(input_file, output_file, "Test")

    with mock.patch("{0}.next".format(builtin_module_name)) as mock_next:
            mock_next.side_effect = Exception("SomeException")

            with pytest.raises(LocopySplitError):
                split_file(input_file, output_file, 2)
            assert not Path("tests/data/mock_output_file.txt.0").exists()
            assert not Path("tests/data/mock_output_file.txt.1").exists()

            with pytest.raises(LocopySplitError):
                split_file(input_file, output_file, 3)
            assert not Path("tests/data/mock_output_file.txt.0").exists()
            assert not Path("tests/data/mock_output_file.txt.1").exists()
            assert not Path("tests/data/mock_output_file.txt.2").exists()