gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: neo4j support in test_delete_default_graph_non_mocked.py

Browse source
This commit is contained in:
Boris Arzentar 2025-11-22 12:44:29 +01:00
parent b8dff548d9
commit 663dd16647
No known key found for this signature in database
GPG key ID: D5CC274C784807B7
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
cognee/tests/test_delete_default_graph_non_mocked.py
View file

@ -1,3 +1,4 @@
import json
import os import os
import pathlib import pathlib
@ -58,7 +59,12 @@ async def main():
for node in initial_nodes: for node in initial_nodes:
node_data = node[1] node_data = node[1]
collection_name = node_data["type"] + "_" + node_data["metadata"]["index_fields"][0] node_metadata = (
node_data["metadata"]
if type(node_data["metadata"]) is dict
else json.loads(node_data["metadata"])
)
collection_name = node_data["type"] + "_" + node_metadata["index_fields"][0]
if collection_name not in initial_nodes_by_vector_collection: if collection_name not in initial_nodes_by_vector_collection:
initial_nodes_by_vector_collection[collection_name] = [] initial_nodes_by_vector_collection[collection_name] = []
initial_nodes_by_vector_collection[collection_name].append(node) initial_nodes_by_vector_collection[collection_name].append(node)
@ -83,7 +89,12 @@ async def main():
after_delete_nodes_by_vector_collection = {} after_delete_nodes_by_vector_collection = {}
for node in initial_nodes: for node in initial_nodes:
node_data = node[1] node_data = node[1]
collection_name = node_data["type"] + "_" + node_data["metadata"]["index_fields"][0] node_metadata = (
node_data["metadata"]
if type(node_data["metadata"]) is dict
else json.loads(node_data["metadata"])
)
collection_name = node_data["type"] + "_" + node_metadata["index_fields"][0]
if collection_name not in after_delete_nodes_by_vector_collection: if collection_name not in after_delete_nodes_by_vector_collection:
after_delete_nodes_by_vector_collection[collection_name] = [] after_delete_nodes_by_vector_collection[collection_name] = []
after_delete_nodes_by_vector_collection[collection_name].append(node) after_delete_nodes_by_vector_collection[collection_name].append(node)