From 0441e19bc9a294761cce95a6c2aaf2f6ec7c8918 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Tue, 26 Nov 2024 16:42:35 +0100
Subject: [PATCH] feat: Adds bruteforce retriever test for neo4j

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

diff --git a/cognee/tests/test_neo4j.py b/cognee/tests/test_neo4j.py
index 60f006c35..87dd789a6 100644
--- a/cognee/tests/test_neo4j.py
+++ b/cognee/tests/test_neo4j.py
@@ -62,6 +62,9 @@ async def main():
 
     assert len(history) == 6, "Search history is not correct."
 
+    results = await brute_force_triplet_search('Who has the most experience with graphic design?')
+    assert len(results) > 0
+
     await cognee.prune.prune_data()
     assert not os.path.isdir(data_directory_path), "Local data files are not deleted"