From 1c73ac03e412b0ae338655285e4e7aab30283006 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Sun, 24 Jul 2022 20:28:24 +0200 Subject: [PATCH] Fix tests --- tests/utilities/test_evaluate_ranking.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/utilities/test_evaluate_ranking.py b/tests/utilities/test_evaluate_ranking.py index b0ef4ad..fd29103 100644 --- a/tests/utilities/test_evaluate_ranking.py +++ b/tests/utilities/test_evaluate_ranking.py @@ -16,7 +16,7 @@ def test_default() -> None: reverse_order=True, ) - assert results == {"d": 1.0, "c": 0.6666666666666666, "b": 0.4} + assert results == {"d": 5 / 6, "c": 2 / 3, "b": 2 / 5} results = evaluate_ranking( ["a", "a", "b", "b", "c", "d", "d", "d"], @@ -26,11 +26,7 @@ def test_default() -> None: disable_interpolation=True, ) - assert results == { - "a": 0.6666666666666666, - "b": 0.3333333333333333, - "c": 0.2857142857142857, - } + assert results == {"a": 0.6, "b": 0.4, "c": 0.2} def test_mismatching_lengths() -> None: