from pathlib import Path
import sys

ROOT = Path(__file__).resolve().parent.parent


def test_topk_tuples_to_ui_items_rounding() -> None:
    sys.path.insert(0, str(ROOT / "src"))
    from predict_format import topk_tuples_to_ui_items

    out = topk_tuples_to_ui_items([("a", 0.123456789), ("b", 0.5)])
    assert out == [{"label": "a", "prob": 0.1235}, {"label": "b", "prob": 0.5}]