fix: Adjust unit test for similarity search

philippe2803 · Mar 5, 2024 · e85a8ae · e85a8ae
1 parent 90792a8
commit e85a8ae
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_vss.py b/tests/test_vss.py
@@ -45,6 +45,6 @@ def test_similarity_search(self):
         cm_vss.load()
         data = cm_vss.similarity_search(query="who is Mistral ai company?", k=2)
         assert len(data) == 2
-        metadatas = [doc.metadata for doc in data]
-        for meta in metadatas:
-            assert meta.get("url") == "https://philippeoger.com/pages/ai-scene-in-europe-last-week/"
+        urls = [doc["url"] for doc in data]
+        for url in urls:
+            assert url == "https://philippeoger.com/pages/ai-scene-in-europe-last-week/"