From 99dc35f23e26e4cd2016f50a6c783f6a0a1749e1 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Wed, 15 Oct 2025 20:01:09 +0200 Subject: [PATCH] fix: resolve issue with neo4j metrics test --- .../tasks/descriptive_metrics/metrics_test_utils.py | 2 -- .../tasks/descriptive_metrics/neo4j_metrics_test.py | 11 ++++++++--- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/cognee/tests/tasks/descriptive_metrics/metrics_test_utils.py b/cognee/tests/tasks/descriptive_metrics/metrics_test_utils.py index 911d9c33b..579a499fd 100644 --- a/cognee/tests/tasks/descriptive_metrics/metrics_test_utils.py +++ b/cognee/tests/tasks/descriptive_metrics/metrics_test_utils.py @@ -1,7 +1,6 @@ from typing import List from cognee.infrastructure.engine import DataPoint from cognee.tasks.storage.add_data_points import add_data_points -from cognee.infrastructure.databases.graph.get_graph_engine import create_graph_engine import cognee from cognee.infrastructure.databases.graph import get_graph_engine import json @@ -64,7 +63,6 @@ async def create_connected_test_graph(): async def get_metrics(provider: str, include_optional=True): - create_graph_engine.cache_clear() cognee.config.set_graph_database_provider(provider) graph_engine = await get_graph_engine() await graph_engine.delete_graph() diff --git a/cognee/tests/tasks/descriptive_metrics/neo4j_metrics_test.py b/cognee/tests/tasks/descriptive_metrics/neo4j_metrics_test.py index 2ca9e9f7e..8d7a6ab02 100644 --- a/cognee/tests/tasks/descriptive_metrics/neo4j_metrics_test.py +++ b/cognee/tests/tasks/descriptive_metrics/neo4j_metrics_test.py @@ -1,7 +1,12 @@ -from cognee.tests.tasks.descriptive_metrics.metrics_test_utils import assert_metrics import asyncio +async def main(): + from cognee.tests.tasks.descriptive_metrics.metrics_test_utils import assert_metrics + + await assert_metrics(provider="neo4j", include_optional=False) + await assert_metrics(provider="neo4j", include_optional=True) + + if __name__ == "__main__": - asyncio.run(assert_metrics(provider="neo4j", include_optional=False)) - asyncio.run(assert_metrics(provider="neo4j", include_optional=True)) + asyncio.run(main())