import os

from vector_utils import utils


def test_write_file_to_disk():
    file_path = './'
    file_name = 'test_write_file.txt'
    file_data = 'line one\r\nline two\n'

    utils.write_file_to_disk(
        file_path, file_name, file_data)

    with open(os.path.join(file_path, file_name), 'rb') as file:
        expected = file.read()

    os.remove(file_path + file_name)

    assert file_data.encode() == expected