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

Fix tests (#663)

Browse source
This commit is contained in:
cybermaggedon 2026-03-06 12:40:02 +00:00 • committed by GitHub
parent 2b9232917c
commit be358efe67
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 27 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/unit/test_storage/test_graph_embeddings_milvus_storage.py
View file

@ -172,26 +172,30 @@ class TestMilvusGraphEmbeddingsStorageProcessor:
message.metadata = MagicMock()
message.metadata.user = 'test_user'
message.metadata.collection = 'test_collection'
valid_entity = EntityEmbeddings(
entity=Term(type=IRI, iri='http://example.com/valid'),
vectors=[[0.1, 0.2, 0.3]]
vectors=[[0.1, 0.2, 0.3]],
chunk_id=''
)
empty_entity = EntityEmbeddings(
entity=Term(type=LITERAL, value=''),
vectors=[[0.4, 0.5, 0.6]]
vectors=[[0.4, 0.5, 0.6]],
chunk_id=''
)
none_entity = EntityEmbeddings(
entity=Term(type=LITERAL, value=None),
vectors=[[0.7, 0.8, 0.9]]
vectors=[[0.7, 0.8, 0.9]],
chunk_id=''
)
message.entities = [valid_entity, empty_entity, none_entity]
await processor.store_graph_embeddings(message)
# Verify only valid entity was inserted with user/collection parameters
# Verify only valid entity was inserted with user/collection/chunk_id parameters
processor.vecstore.insert.assert_called_once_with(
[0.1, 0.2, 0.3], 'http://example.com/valid', 'test_user', 'test_collection'
[0.1, 0.2, 0.3], 'http://example.com/valid', 'test_user', 'test_collection',
chunk_id=''
)
@pytest.mark.asyncio