apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-28 18:06:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Updating tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-09 21:38:10 +01:00
parent 7da94add4b
commit b44be9c5d7
3 changed files with 141 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/unit/test_query/test_doc_embeddings_milvus_query.py
View file

@ -85,8 +85,10 @@ class TestMilvusDocEmbeddingsQueryProcessor:
result = await processor.query_document_embeddings(query)
# Verify search was called with correct parameters
processor.vecstore.search.assert_called_once_with([0.1, 0.2, 0.3], limit=5)
# Verify search was called with correct parameters including user/collection
processor.vecstore.search.assert_called_once_with(
[0.1, 0.2, 0.3], 'test_user', 'test_collection', limit=5
)
# Verify results are document chunks
assert len(result) == 3