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

updated tests

Browse source
This commit is contained in:
Vasilije 2024-06-01 21:43:29 +02:00
parent 0f8ec35d4c
commit b7df2dacc7
1 changed files with 25 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
cognee/tests/test_library.py
View file

@ -45,6 +45,31 @@ async def main():
Some notable LLMs are OpenAI's GPT series of models (e.g., GPT-3.5 and GPT-4, used in ChatGPT and Microsoft Copilot), Google's PaLM and Gemini (the latter of which is currently used in the chatbot of the same name), xAI's Grok, Meta's LLaMA family of open-source models, Anthropic's Claude models, Mistral AI's open source models, and Databricks' open source DBRX.
"""
await cognee.cognify([ "cs_explanations"])
search_results = cognee.search("SIMILARITY", "computer science")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = cognee.search("CATEGORIES", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = cognee.search("NEIGHBOR", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = cognee.search("SUMMARY", "Work and computers")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
search_results = cognee.search("ADJACENT", "DefaultGraphModel__default_user")
assert len(search_results) != 0, "The search results list is empty."
print("The search results list is not empty.")
#
# dataset_name = "cs_explanations"
# await cognee.add(
@ -55,15 +80,6 @@ async def main():
# dataset_name
# )
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.")