"""Expected clustering result for k-mean tests."""

k_means_expected_results = {
    "mean_result": [
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        1,
        0,
        1,
        0,
        1,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
    ],
    "dba_result": [1, 0, 1, 0, 0],
    "mean_train_score": 0.47307692307692306,
    "dba_train_score": 0.6,
    "mean_test_score": 0.3192307692307692,
    "dba_test_score": 0.4,
    "mean_iters": 6,
    "dba_iters": 3,
    "mean_labels": [
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        1,
        1,
        0,
        0,
        0,
        1,
        1,
        0,
        1,
        1,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        0,
        1,
        0,
        0,
        1,
        1,
        0,
        0,
    ],
    "dba_labels": [0, 1, 0, 0, 0],
}
