From 5e67d49bd534561962722e03151a2ad72e12ba84 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 16 Feb 2026 17:37:02 +0000 Subject: [PATCH] Update tests --- tests/integration/test_kg_extract_store_integration.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/integration/test_kg_extract_store_integration.py b/tests/integration/test_kg_extract_store_integration.py index 9131f1ee..2baa1d4d 100644 --- a/tests/integration/test_kg_extract_store_integration.py +++ b/tests/integration/test_kg_extract_store_integration.py @@ -147,6 +147,8 @@ class TestKnowledgeGraphPipelineIntegration: processor.emit_triples = DefinitionsProcessor.emit_triples.__get__(processor, DefinitionsProcessor) processor.emit_ecs = DefinitionsProcessor.emit_ecs.__get__(processor, DefinitionsProcessor) processor.on_message = DefinitionsProcessor.on_message.__get__(processor, DefinitionsProcessor) + processor.triples_batch_size = 50 + processor.entity_batch_size = 5 return processor @pytest.fixture @@ -156,6 +158,7 @@ class TestKnowledgeGraphPipelineIntegration: processor.to_uri = RelationshipsProcessor.to_uri.__get__(processor, RelationshipsProcessor) processor.emit_triples = RelationshipsProcessor.emit_triples.__get__(processor, RelationshipsProcessor) processor.on_message = RelationshipsProcessor.on_message.__get__(processor, RelationshipsProcessor) + processor.triples_batch_size = 50 return processor @pytest.mark.asyncio