From 746681837e5ed4a8f16d2dc64c03e27271a1d491 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Wed, 12 Jun 2024 23:40:53 +0200
Subject: [PATCH] Add NEO4J test

---
 cognee/tests/test_neo4j.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/cognee/tests/test_neo4j.py b/cognee/tests/test_neo4j.py
index ba71ba9be..b25519059 100644
--- a/cognee/tests/test_neo4j.py
+++ b/cognee/tests/test_neo4j.py
@@ -21,6 +21,8 @@ async def main():
 
     cognee.config.set_graph_database_provider("neo4j")
 
+    await cognee.prune.prune_system()
+
     logging.debug("CURRENCT CWD: %s", pathlib.Path(__file__).parent)
     logging.debug("CURRENCT CWD: %s", os.getcwd())
 
@@ -49,9 +51,9 @@ async def main():
 
     await cognee.cognify(["cs_explanations"])
 
-    # search_results = await cognee.search("SIMILARITY", {"query": "computer science"})
-    # assert len(search_results) != 0, "The search results list is empty."
-    # print("The search results list is not empty.")
+    search_results = await cognee.search("SIMILARITY", {"query": "computer science"})
+    assert len(search_results) != 0, "The search results list is empty."
+    print("The search results list is not empty.")
     #
     # search_results = await cognee.search("CATEGORIES", {"query": "DefaultGraphModel__default_user"})
     # assert len(search_results) != 0, "The search results list is empty."