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

chapter3/retrieval-pipeline/test_weighted_fusion_dedup.py

@@ -0,0 +1,15 @@
+import pytest
+from fusion import weighted_score_fusion
+
+def test_weighted_score_fusion_preserves_top_score_on_duplicate_doc_id():
+    """Verify weighted score fusion preserves top score when duplicate doc_ids exist in ranked list."""
+    ranked_lists = {
+        "dense": [("doc1", 0.95), ("doc2", 0.80), ("doc1", 0.10)],
+        "sparse": [("doc1", 10.0), ("doc2", 5.0)]
+    }
+
+    results = weighted_score_fusion(ranked_lists)
+
+    assert results[0][0] == "doc1"
+    assert results[1][0] == "doc2"
+    assert results[0][1] > results[1][1]