apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 12:11:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-01-27 00:44:10 +00:00
parent 745ac4dac7
commit 06360fcb43
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -136,7 +136,7 @@ class TestMilvusGraphEmbeddingsStorageProcessor:
message.metadata.collection = 'test_collection' message.metadata.collection = 'test_collection'
entity = EntityEmbeddings( entity = EntityEmbeddings(
entity=Value(value='', is_uri=False), entity=Term(type=LITERAL, value=''),
vectors=[[0.1, 0.2, 0.3]] vectors=[[0.1, 0.2, 0.3]]
) )
message.entities = [entity] message.entities = [entity]
@ -155,7 +155,7 @@ class TestMilvusGraphEmbeddingsStorageProcessor:
message.metadata.collection = 'test_collection' message.metadata.collection = 'test_collection'
entity = EntityEmbeddings( entity = EntityEmbeddings(
entity=Value(value=None, is_uri=False), entity=Term(type=LITERAL, value=None),
vectors=[[0.1, 0.2, 0.3]] vectors=[[0.1, 0.2, 0.3]]
) )
message.entities = [entity] message.entities = [entity]
@ -174,15 +174,15 @@ class TestMilvusGraphEmbeddingsStorageProcessor:
message.metadata.collection = 'test_collection' message.metadata.collection = 'test_collection'
valid_entity = EntityEmbeddings( valid_entity = EntityEmbeddings(
entity=Value(value='http://example.com/valid', is_uri=True), entity=Term(type=IRI, iri='http://example.com/valid'),
vectors=[[0.1, 0.2, 0.3]] vectors=[[0.1, 0.2, 0.3]]
) )
empty_entity = EntityEmbeddings( empty_entity = EntityEmbeddings(
entity=Value(value='', is_uri=False), entity=Term(type=LITERAL, value=''),
vectors=[[0.4, 0.5, 0.6]] vectors=[[0.4, 0.5, 0.6]]
) )
none_entity = EntityEmbeddings( none_entity = EntityEmbeddings(
entity=Value(value=None, is_uri=False), entity=Term(type=LITERAL, value=None),
vectors=[[0.7, 0.8, 0.9]] vectors=[[0.7, 0.8, 0.9]]
) )
message.entities = [valid_entity, empty_entity, none_entity] message.entities = [valid_entity, empty_entity, none_entity]
@ -217,7 +217,7 @@ class TestMilvusGraphEmbeddingsStorageProcessor:
message.metadata.collection = 'test_collection' message.metadata.collection = 'test_collection'
entity = EntityEmbeddings( entity = EntityEmbeddings(
entity=Value(value='http://example.com/entity', is_uri=True), entity=Term(type=IRI, iri='http://example.com/entity'),
vectors=[] vectors=[]
) )
message.entities = [entity] message.entities = [entity]
@ -236,7 +236,7 @@ class TestMilvusGraphEmbeddingsStorageProcessor:
message.metadata.collection = 'test_collection' message.metadata.collection = 'test_collection'
entity = EntityEmbeddings( entity = EntityEmbeddings(
entity=Value(value='http://example.com/entity', is_uri=True), entity=Term(type=IRI, iri='http://example.com/entity'),
vectors=[ vectors=[
[0.1, 0.2], # 2D vector [0.1, 0.2], # 2D vector
[0.3, 0.4, 0.5, 0.6], # 4D vector [0.3, 0.4, 0.5, 0.6], # 4D vector
@ -269,11 +269,11 @@ class TestMilvusGraphEmbeddingsStorageProcessor:
message.metadata.collection = 'test_collection' message.metadata.collection = 'test_collection'
uri_entity = EntityEmbeddings( uri_entity = EntityEmbeddings(
entity=Value(value='http://example.com/uri_entity', is_uri=True), entity=Term(type=IRI, iri='http://example.com/uri_entity'),
vectors=[[0.1, 0.2, 0.3]] vectors=[[0.1, 0.2, 0.3]]
) )
literal_entity = EntityEmbeddings( literal_entity = EntityEmbeddings(
entity=Value(value='literal entity text', is_uri=False), entity=Term(type=LITERAL, value='literal entity text'),
vectors=[[0.4, 0.5, 0.6]] vectors=[[0.4, 0.5, 0.6]]
) )
message.entities = [uri_entity, literal_entity] message.entities = [uri_entity, literal_entity]