from atlas_um.helpers.csv import dump_data_to_csv


def test_dump_data_to_csv():
    test_headers = ["col1", "col2", "col3"]
    test_rows = [
        ("col1_value", "col2_value", "col3_value"),
        ("col1_value", "col2_value", "col3_value"),
    ]
    expected_result = (
        "col1,col2,col3\r\n"
        "col1_value,col2_value,col3_value\r\n"
        "col1_value,col2_value,col3_value\r\n"
    )

    actual_result = dump_data_to_csv(test_rows, test_headers)
    assert expected_result == actual_result