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

fix: await search in tests

Browse source
This commit is contained in:
Boris Arzentar 2024-06-03 15:04:26 +02:00
parent 8499b7f2fc
commit 7dc36bb4c7
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cognee/tests/test_library.py
View file

@ -48,25 +48,25 @@ async def main():
await cognee.cognify([ "cs_explanations"]) await cognee.cognify([ "cs_explanations"])
search_results = cognee.search("SIMILARITY", "computer science") search_results = await cognee.search("SIMILARITY", "computer science")
assert len(search_results) != 0, "The search results list is empty." assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.") print("The search results list is not empty.")
search_results = cognee.search("CATEGORIES", "DefaultGraphModel__default_user") search_results = await cognee.search("CATEGORIES", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty." assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.") print("The search results list is not empty.")
search_results = cognee.search("NEIGHBOR", "DefaultGraphModel__default_user") search_results = await cognee.search("NEIGHBOR", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty." assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.") print("The search results list is not empty.")
search_results = cognee.search("SUMMARY", "Work and computers") search_results = await cognee.search("SUMMARY", "Work and computers")
assert len(search_results) != 0, "The search results list is empty." assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.") print("The search results list is not empty.")
search_results = cognee.search("ADJACENT", "DefaultGraphModel__default_user") search_results = await cognee.search("ADJACENT", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty." assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.") print("The search results list is not empty.")