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

fix: wrap search query in tests

Browse source
This commit is contained in:
Boris Arzentar 2024-06-03 15:10:45 +02:00
parent 7dc36bb4c7
commit e9d0bf8bd4
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"])
search_results = await cognee.search("SIMILARITY", "computer science")
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", "DefaultGraphModel__default_user")
search_results = await cognee.search("CATEGORIES", { "query": "DefaultGraphModel__default_user" })
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = await cognee.search("NEIGHBOR", "DefaultGraphModel__default_user")
search_results = await cognee.search("NEIGHBOR", { "query": "DefaultGraphModel__default_user" })
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = await cognee.search("SUMMARY", "Work and computers")
search_results = await cognee.search("SUMMARY", { "query": "Work and computers" })
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = await cognee.search("ADJACENT", "DefaultGraphModel__default_user")
search_results = await cognee.search("ADJACENT", { "query": "DefaultGraphModel__default_user" })
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")