klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Qdrant Tests (#2287)

Browse Source
This commit is contained in:
Dev Khant
2025-03-03 10:46:56 +05:30
committed by GitHub
parent f4dc5f6c71
commit a4340b2336
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/vector_stores/test_qdrant.py
View File

@@ -51,7 +51,8 @@ class TestQdrant(unittest.TestCase):
def test_search(self): def test_search(self):
query_vector = [0.1, 0.2] query_vector = [0.1, 0.2]
self.client_mock.search.return_value = [{"id": str(uuid.uuid4()), "score": 0.95, "payload": {"key": "value"}}] mock_point = MagicMock(id=str(uuid.uuid4()), score=0.95, payload={"key": "value"})
self.client_mock.query_points.return_value = MagicMock(points=[mock_point])
results = self.qdrant.search(query=query_vector, limit=1) results = self.qdrant.search(query=query_vector, limit=1)
@@ -63,9 +64,8 @@ class TestQdrant(unittest.TestCase):
) )
self.assertEqual(len(results), 1) self.assertEqual(len(results), 1)
self.assertIn("id", results[0]) self.assertEqual(results[0].payload, {"key": "value"})
self.assertIn("score", results[0]) self.assertEqual(results[0].score, 0.95)
self.assertIn("payload", results[0])
def test_delete(self): def test_delete(self):
vector_id = str(uuid.uuid4()) vector_id = str(uuid.uuid4())