diff --git a/tests/test_evals/test_utils/test_logging.py b/tests/test_evals/test_utils/test_logging.py index 2543726..4116114 100644 --- a/tests/test_evals/test_utils/test_logging.py +++ b/tests/test_evals/test_utils/test_logging.py @@ -463,6 +463,18 @@ def test_log_caching(): metric = unify.get_logs_metric(project=project, metric="mean", key="b") assert metric == 2.5 + # get_groups + unify.get_groups(project=project, key="b") + assert ( + _get_cache( + fn_name="get_groups", + kw={"project": project, "key": "b"}, + ) + is not None + ) + groups = unify.get_groups(project=project, key="b") + assert groups == {"0": 2, "1": 3} + if __name__ == "__main__": pass