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

Fixing tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-01-27 09:54:12 +00:00
parent dc3f5d6db3
commit 96441938e3
2 changed files with 19 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/unit/test_query/test_graph_embeddings_milvus_query.py
View file

@ -70,7 +70,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor:
result = processor.create_value("http://example.com/resource") result = processor.create_value("http://example.com/resource")
assert isinstance(result, Term) assert isinstance(result, Term)
assert result.value == "http://example.com/resource" assert result.iri == "http://example.com/resource"
assert result.type == IRI assert result.type == IRI
def test_create_value_with_https_uri(self, processor): def test_create_value_with_https_uri(self, processor):
@ -78,7 +78,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor:
result = processor.create_value("https://example.com/resource") result = processor.create_value("https://example.com/resource")
assert isinstance(result, Term) assert isinstance(result, Term)
assert result.value == "https://example.com/resource" assert result.iri == "https://example.com/resource"
assert result.type == IRI assert result.type == IRI
def test_create_value_with_literal(self, processor): def test_create_value_with_literal(self, processor):
@ -138,13 +138,13 @@ class TestMilvusGraphEmbeddingsQueryProcessor:
[0.1, 0.2, 0.3], 'test_user', 'test_collection', limit=10 [0.1, 0.2, 0.3], 'test_user', 'test_collection', limit=10
) )
# Verify results are converted to Value objects # Verify results are converted to Term objects
assert len(result) == 3 assert len(result) == 3
assert isinstance(result[0], Term) assert isinstance(result[0], Term)
assert result[0].value == "http://example.com/entity1" assert result[0].iri == "http://example.com/entity1"
assert result[0].type == IRI assert result[0].type == IRI
assert isinstance(result[1], Term) assert isinstance(result[1], Term)
assert result[1].value == "http://example.com/entity2" assert result[1].iri == "http://example.com/entity2"
assert result[1].type == IRI assert result[1].type == IRI
assert isinstance(result[2], Term) assert isinstance(result[2], Term)
assert result[2].value == "literal entity" assert result[2].value == "literal entity"
@ -186,7 +186,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor:
# Verify results are deduplicated and limited # Verify results are deduplicated and limited
assert len(result) == 3 assert len(result) == 3
entity_values = [r.value for r in result] entity_values = [r.iri if r.type == IRI else r.value for r in result]
assert "http://example.com/entity1" in entity_values assert "http://example.com/entity1" in entity_values
assert "http://example.com/entity2" in entity_values assert "http://example.com/entity2" in entity_values
assert "http://example.com/entity3" in entity_values assert "http://example.com/entity3" in entity_values
@ -246,7 +246,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor:
# Verify duplicates are removed # Verify duplicates are removed
assert len(result) == 3 assert len(result) == 3
entity_values = [r.value for r in result] entity_values = [r.iri if r.type == IRI else r.value for r in result]
assert len(set(entity_values)) == 3 # All unique assert len(set(entity_values)) == 3 # All unique
assert "http://example.com/entity1" in entity_values assert "http://example.com/entity1" in entity_values
assert "http://example.com/entity2" in entity_values assert "http://example.com/entity2" in entity_values
@ -348,7 +348,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor:
# Check URI entities # Check URI entities
uri_results = [r for r in result if r.type == IRI] uri_results = [r for r in result if r.type == IRI]
assert len(uri_results) == 2 assert len(uri_results) == 2
uri_values = [r.value for r in uri_results] uri_values = [r.iri for r in uri_results]
assert "http://example.com/uri_entity" in uri_values assert "http://example.com/uri_entity" in uri_values
assert "https://example.com/another_uri" in uri_values assert "https://example.com/another_uri" in uri_values
@ -486,7 +486,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor:
# Verify results from all dimensions # Verify results from all dimensions
assert len(result) == 3 assert len(result) == 3
entity_values = [r.value for r in result] entity_values = [r.iri if r.type == IRI else r.value for r in result]
assert "entity_2d" in entity_values assert "entity_2d" in entity_values
assert "entity_4d" in entity_values assert "entity_4d" in entity_values
assert "entity_3d" in entity_values assert "entity_3d" in entity_values

16
tests/unit/test_query/test_graph_embeddings_qdrant_query.py
View file

@ -85,10 +85,10 @@ class TestQdrantGraphEmbeddingsQuery(IsolatedAsyncioTestCase):
value = processor.create_value('http://example.com/entity') value = processor.create_value('http://example.com/entity')
# Assert # Assert
assert hasattr(value, 'value') assert hasattr(value, 'iri')
assert value.value == 'http://example.com/entity' assert value.iri == 'http://example.com/entity'
assert hasattr(value, 'is_uri') assert hasattr(value, 'type')
assert value.is_uri == True assert value.type == IRI
@patch('trustgraph.query.graph_embeddings.qdrant.service.QdrantClient') @patch('trustgraph.query.graph_embeddings.qdrant.service.QdrantClient')
@patch('trustgraph.base.GraphEmbeddingsQueryService.__init__') @patch('trustgraph.base.GraphEmbeddingsQueryService.__init__')
@ -109,10 +109,10 @@ class TestQdrantGraphEmbeddingsQuery(IsolatedAsyncioTestCase):
value = processor.create_value('https://secure.example.com/entity') value = processor.create_value('https://secure.example.com/entity')
# Assert # Assert
assert hasattr(value, 'value') assert hasattr(value, 'iri')
assert value.value == 'https://secure.example.com/entity' assert value.iri == 'https://secure.example.com/entity'
assert hasattr(value, 'is_uri') assert hasattr(value, 'type')
assert value.is_uri == True assert value.type == IRI
@patch('trustgraph.query.graph_embeddings.qdrant.service.QdrantClient') @patch('trustgraph.query.graph_embeddings.qdrant.service.QdrantClient')
@patch('trustgraph.base.GraphEmbeddingsQueryService.__init__') @patch('trustgraph.base.GraphEmbeddingsQueryService.__init__')