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

Add NEO4J test

Browse source
This commit is contained in:
Vasilije 2024-06-12 23:40:53 +02:00
parent adedfa41a0
commit 746681837e
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/tests/test_neo4j.py
View file

@ -21,6 +21,8 @@ async def main():
cognee.config.set_graph_database_provider("neo4j") cognee.config.set_graph_database_provider("neo4j")
await cognee.prune.prune_system()
logging.debug("CURRENCT CWD: %s", pathlib.Path(__file__).parent) logging.debug("CURRENCT CWD: %s", pathlib.Path(__file__).parent)
logging.debug("CURRENCT CWD: %s", os.getcwd()) logging.debug("CURRENCT CWD: %s", os.getcwd())
@ -49,9 +51,9 @@ async def main():
await cognee.cognify(["cs_explanations"]) await cognee.cognify(["cs_explanations"])
# search_results = await cognee.search("SIMILARITY", {"query": "computer science"}) search_results = await cognee.search("SIMILARITY", {"query": "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 = await cognee.search("CATEGORIES", {"query": "DefaultGraphModel__default_user"}) # search_results = await cognee.search("CATEGORIES", {"query": "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."