From 96441938e320ee13332fb208ba1dbfd27965bd8c Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 27 Jan 2026 09:54:12 +0000 Subject: [PATCH] Fixing tests --- .../test_graph_embeddings_milvus_query.py | 22 +++++++++---------- .../test_graph_embeddings_qdrant_query.py | 16 +++++++------- 2 files changed, 19 insertions(+), 19 deletions(-) diff --git a/tests/unit/test_query/test_graph_embeddings_milvus_query.py b/tests/unit/test_query/test_graph_embeddings_milvus_query.py index 7bd096fe..21b6e1bf 100644 --- a/tests/unit/test_query/test_graph_embeddings_milvus_query.py +++ b/tests/unit/test_query/test_graph_embeddings_milvus_query.py @@ -68,17 +68,17 @@ class TestMilvusGraphEmbeddingsQueryProcessor: def test_create_value_with_http_uri(self, processor): """Test create_value with HTTP URI""" result = processor.create_value("http://example.com/resource") - + assert isinstance(result, Term) - assert result.value == "http://example.com/resource" + assert result.iri == "http://example.com/resource" assert result.type == IRI def test_create_value_with_https_uri(self, processor): """Test create_value with HTTPS URI""" result = processor.create_value("https://example.com/resource") - + assert isinstance(result, Term) - assert result.value == "https://example.com/resource" + assert result.iri == "https://example.com/resource" assert result.type == IRI 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 ) - # Verify results are converted to Value objects + # Verify results are converted to Term objects assert len(result) == 3 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 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 isinstance(result[2], Term) assert result[2].value == "literal entity" @@ -186,7 +186,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor: # Verify results are deduplicated and limited 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/entity2" in entity_values assert "http://example.com/entity3" in entity_values @@ -246,7 +246,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor: # Verify duplicates are removed 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 "http://example.com/entity1" in entity_values assert "http://example.com/entity2" in entity_values @@ -348,7 +348,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor: # Check URI entities uri_results = [r for r in result if r.type == IRI] 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 "https://example.com/another_uri" in uri_values @@ -486,7 +486,7 @@ class TestMilvusGraphEmbeddingsQueryProcessor: # Verify results from all dimensions 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_4d" in entity_values assert "entity_3d" in entity_values \ No newline at end of file diff --git a/tests/unit/test_query/test_graph_embeddings_qdrant_query.py b/tests/unit/test_query/test_graph_embeddings_qdrant_query.py index ab22c9df..ab3498b3 100644 --- a/tests/unit/test_query/test_graph_embeddings_qdrant_query.py +++ b/tests/unit/test_query/test_graph_embeddings_qdrant_query.py @@ -85,10 +85,10 @@ class TestQdrantGraphEmbeddingsQuery(IsolatedAsyncioTestCase): value = processor.create_value('http://example.com/entity') # Assert - assert hasattr(value, 'value') - assert value.value == 'http://example.com/entity' - assert hasattr(value, 'is_uri') - assert value.is_uri == True + assert hasattr(value, 'iri') + assert value.iri == 'http://example.com/entity' + assert hasattr(value, 'type') + assert value.type == IRI @patch('trustgraph.query.graph_embeddings.qdrant.service.QdrantClient') @patch('trustgraph.base.GraphEmbeddingsQueryService.__init__') @@ -109,10 +109,10 @@ class TestQdrantGraphEmbeddingsQuery(IsolatedAsyncioTestCase): value = processor.create_value('https://secure.example.com/entity') # Assert - assert hasattr(value, 'value') - assert value.value == 'https://secure.example.com/entity' - assert hasattr(value, 'is_uri') - assert value.is_uri == True + assert hasattr(value, 'iri') + assert value.iri == 'https://secure.example.com/entity' + assert hasattr(value, 'type') + assert value.type == IRI @patch('trustgraph.query.graph_embeddings.qdrant.service.QdrantClient') @patch('trustgraph.base.GraphEmbeddingsQueryService.__init__')