File size: 384 Bytes
179dfc2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | 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}]
|