Improved reranking test

ankane · Aug 30, 2024 · 9789d33 · 9789d33
1 parent 2a1837e
commit 9789d33
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/pipeline_test.rb b/test/pipeline_test.rb
@@ -67,11 +67,15 @@ def test_embedding
   end
 
   def test_reranking
-    ranker = Informers.pipeline("reranking")
-    result = ranker.("Who created Ruby?", ["Matz created Ruby", "Another doc"])
+    query = "How many people live in London?"
+    docs = ["Around 9 Million people live in London", "London is known for its financial district"]
+    rerank = Informers.pipeline("reranking")
+    result = rerank.(query, docs)
     assert_equal 2, result.size
     assert_equal 0, result[0][:doc_id]
+    assert_in_delta 0.984, result[0][:score]
     assert_equal 1, result[1][:doc_id]
+    assert_in_delta 0.139, result[1][:score]
   end
 
   def test_progress_callback