"""Tests for common utilities."""

from packages.core.common.utils import get_distance_bucket


def test_get_distance_bucket_fixed_mode():
    """Test fixed-size distance buckets."""
    assert (
        get_distance_bucket(0, [1700, 2000], mode="fixed", bucket_size=400) == "0-399"
    )
    assert (
        get_distance_bucket(400, [1700, 2000], mode="fixed", bucket_size=400)
        == "400-799"
    )
    assert (
        get_distance_bucket(999, [1700, 2000], mode="fixed", bucket_size=400)
        == "800-1199"
    )


def test_get_distance_bucket_thresholds_mode():
    """Test threshold-based distance buckets."""
    buckets = [1700, 2000, 2400]
    assert get_distance_bucket(1600, buckets) == "<1700"
    assert get_distance_bucket(1800, buckets) == "1700-2000"
    assert get_distance_bucket(2500, buckets) == ">2400"