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

Fix pytests (#2157)

Browse Source
This commit is contained in:
Dev Khant
2025-01-19 04:36:49 +05:30
committed by GitHub
parent 80c9c6a577
commit a9d1383909
1 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/vector_stores/test_elasticsearch.py
View File

@@ -208,15 +208,19 @@ class TestElasticsearchDB(unittest.TestCase):
# Verify search parameters # Verify search parameters
self.assertEqual(search_args["index"], "test_collection") self.assertEqual(search_args["index"], "test_collection")
body = search_args["body"] body = search_args["body"]
self.assertIn("script_score", body["query"])
self.assertEqual( # Verify KNN query structure
body["query"]["script_score"]["script"]["params"]["query_vector"], self.assertIn("query", body)
query_vector self.assertIn("bool", body["query"])
) self.assertIn("must", body["query"]["bool"])
# Verify KNN parameters
knn_query = body["query"]["bool"]["must"][-1]["knn"]["vector"]
self.assertEqual(knn_query["vector"], query_vector)
self.assertEqual(knn_query["k"], 5)
# Verify results # Verify results
self.assertEqual(len(results), 1) self.assertEqual(len(results), 1)
self.assertIsInstance(results[0], OutputData)
self.assertEqual(results[0].id, "id1") self.assertEqual(results[0].id, "id1")
self.assertEqual(results[0].score, 0.8) self.assertEqual(results[0].score, 0.8)
self.assertEqual(results[0].payload, {"key1": "value1"}) self.assertEqual(results[0].payload, {"key1": "value1"})