added get_groups to test_log_caching.

unifyai · Nov 7, 2024 · bd1354e · bd1354e
1 parent 3658f32
commit bd1354e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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