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

Fix tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-02-09 14:56:52 +00:00
parent bb28718daf
commit 16c83c82aa
1 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/integration/test_kg_extract_store_integration.py
View file

@ -16,7 +16,7 @@ from trustgraph.extract.kg.definitions.extract import Processor as DefinitionsPr
from trustgraph.extract.kg.relationships.extract import Processor as RelationshipsProcessor from trustgraph.extract.kg.relationships.extract import Processor as RelationshipsProcessor
from trustgraph.storage.knowledge.store import Processor as KnowledgeStoreProcessor from trustgraph.storage.knowledge.store import Processor as KnowledgeStoreProcessor
from trustgraph.schema import Chunk, Triple, Triples, Metadata, Term, Error, IRI, LITERAL from trustgraph.schema import Chunk, Triple, Triples, Metadata, Term, Error, IRI, LITERAL
from trustgraph.schema import EntityContext, EntityContexts, GraphEmbeddings from trustgraph.schema import EntityContext, EntityContexts, GraphEmbeddings, EntityEmbeddings
from trustgraph.rdf import TRUSTGRAPH_ENTITIES, DEFINITION, RDF_LABEL, SUBJECT_OF from trustgraph.rdf import TRUSTGRAPH_ENTITIES, DEFINITION, RDF_LABEL, SUBJECT_OF
@ -405,7 +405,12 @@ class TestKnowledgeGraphPipelineIntegration:
collection="test_collection", collection="test_collection",
metadata=[] metadata=[]
), ),
entities=[] entities=[
EntityEmbeddings(
entity=Term(type=IRI, iri="http://example.org/entity"),
vectors=[[0.1, 0.2, 0.3]]
)
]
) )
mock_msg = MagicMock() mock_msg = MagicMock()
@ -496,12 +501,12 @@ class TestKnowledgeGraphPipelineIntegration:
await definitions_processor.on_message(mock_msg, mock_consumer, mock_flow_context) await definitions_processor.on_message(mock_msg, mock_consumer, mock_flow_context)
# Assert # Assert
# Should still call producers but with empty results # Should NOT call producers with empty results (avoids Cassandra NULL issues)
triples_producer = mock_flow_context("triples") triples_producer = mock_flow_context("triples")
entity_contexts_producer = mock_flow_context("entity-contexts") entity_contexts_producer = mock_flow_context("entity-contexts")
triples_producer.send.assert_called_once() triples_producer.send.assert_not_called()
entity_contexts_producer.send.assert_called_once() entity_contexts_producer.send.assert_not_called()
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_invalid_extraction_format_handling(self, definitions_processor, mock_flow_context, sample_chunk): async def test_invalid_extraction_format_handling(self, definitions_processor, mock_flow_context, sample_chunk):