From 0f8ec35d4c2255c055bc7c2f97a9f1e39dd1aaeb Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Sat, 1 Jun 2024 18:41:08 +0200
Subject: [PATCH] fix for cognee

---
 cognee/tests/test_library.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/cognee/tests/test_library.py b/cognee/tests/test_library.py
index eaac864bc..3d1a1b922 100755
--- a/cognee/tests/test_library.py
+++ b/cognee/tests/test_library.py
@@ -57,6 +57,20 @@ async def  main():
 
     await cognee.cognify([ "cs_explanations"])
 
+    search_results = cognee.search("SIMILARITY", "computer science")
+    if len(search_results) == 0:
+        print("The similarity search results list is empty.")
+        raise Exception("The search results list is empty.")
+    else:
+        print("The search results list is not empty.")
+
+    # search_results = cognee.search("SIMILARITY", "computer science")
+    # if len(search_results) == 0:
+    #     print("The similarity search results list is empty.")
+    #     raise Exception("The search results list is empty.")
+    # else:
+    #     print("The search results list is not empty.")
+
 
 if __name__ == "__main__":
     import asyncio