ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter7/elo-leaderboard/tests/test_animation_tie_wins.py

@@ -0,0 +1,35 @@
+"""prepare_animation_data must keep fractional wins from Elo ties."""
+import pandas as pd
+from animation import prepare_animation_data
+
+
+def test_tie_half_wins_are_not_truncated():
+    history = pd.DataFrame(
+        {
+            "date": pd.to_datetime(["2024-01-07", "2024-01-07"]),
+            "model": ["A", "B"],
+            "rating": [1000.0, 1000.0],
+            "rank": [1, 2],
+            "matches": [1, 1],
+            "wins": [0.5, 0.5],
+        }
+    )
+    data = prepare_animation_data(history, top_n=2)
+    wins = {m["name"]: m["wins"] for m in data["frames"][0]["models"]}
+    assert wins["A"] == 0.5
+    assert wins["B"] == 0.5
+
+
+def test_whole_wins_still_serialize():
+    history = pd.DataFrame(
+        {
+            "date": pd.to_datetime(["2024-01-07"]),
+            "model": ["A"],
+            "rating": [1010.0],
+            "rank": [1],
+            "matches": [2],
+            "wins": [2.0],
+        }
+    )
+    data = prepare_animation_data(history, top_n=1)
+    assert data["frames"][0]["models"][0]["wins"] == 2.0