from .conftest import EXAMPLE_DIR_WITH_PARAMS

def test_split_data_frame(example_dir):
    from sensospot_data.parser import parse_folder
    from sensospot_data.utils import split_data_frame

    data_frame = parse_folder(example_dir / EXAMPLE_DIR_WITH_PARAMS)

    result = split_data_frame(data_frame, "Well.Row")

    assert set(result.keys()) == set("ABC")
    for key, value_df in result.items():
        assert set(value_df["Well.Row"].unique()) == {key}