"""Tests para distance_bucket."""

import os
import sys

sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "..", ".."))

from python.functions.geo.distance_bucket import distance_bucket


def test_bucket_0_5():
    assert distance_bucket(3.0) == "0-5"


def test_bucket_5_10():
    assert distance_bucket(7.0) == "5-10"


def test_bucket_borde_exacto():
    # 10 <= 10 → "5-10"
    assert distance_bucket(10.0) == "5-10"


def test_bucket_160_mas():
    assert distance_bucket(200.0) == "160+"
