from pyannote_diarization_3_1_mlx._config import (
    SEG_DURATION, SEG_HOP, SEG_FRAMES, SEG_CLASSES,
    MAX_SPEAKERS_PER_CHUNK, MAX_SPEAKERS_PER_FRAME,
    EMB_BATCH_SIZE, EMB_EXCLUDE_OVERLAP,
    CLUSTER_METHOD, CLUSTER_THRESHOLD, CLUSTER_MIN_SIZE,
    SEG_HF_REPO, SEG_HF_REV, EMB_HF_REPO, EMB_HF_REV,
)


def test_pyannote_3_1_locked_hyperparameters():
    assert SEG_DURATION == 10.0
    assert SEG_HOP == 1.0
    assert SEG_FRAMES == 589
    assert SEG_CLASSES == 7
    assert MAX_SPEAKERS_PER_CHUNK == 3
    assert MAX_SPEAKERS_PER_FRAME == 2
    assert EMB_BATCH_SIZE == 32
    assert EMB_EXCLUDE_OVERLAP is True
    assert CLUSTER_METHOD == "centroid"
    assert CLUSTER_THRESHOLD == 0.7045654963945799
    assert CLUSTER_MIN_SIZE == 12


def test_locked_hf_revisions():
    assert SEG_HF_REPO == "mlx-community/pyannote-segmentation-3.0-mlx"
    assert SEG_HF_REV == "5189a69b35c5f7e48082a978f3476bac81590874"
    assert EMB_HF_REPO == "mlx-community/wespeaker-voxceleb-resnet34-LM"
    assert EMB_HF_REV == "97fc9343d2cfd0ae4d1c1d8c299e0046aa502e31"