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

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

diff --git a/cognee/tests/test_neo4j.py b/cognee/tests/test_neo4j.py
index 756b29cc4..8eea5b7f7 100644
--- a/cognee/tests/test_neo4j.py
+++ b/cognee/tests/test_neo4j.py
@@ -4,6 +4,7 @@ import logging
 import pathlib
 import cognee
 from cognee.api.v1.search import SearchType
+from cognee.modules.retrieval.brute_force_triplet_search import brute_force_triplet_search
 
 logging.basicConfig(level=logging.DEBUG)
 
@@ -61,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
+
 if __name__ == "__main__":
     import asyncio
     asyncio.run(main())