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

feat: Adds tests for pgvector, qdrant and weaviate

Browse source
This commit is contained in:
hajdul88 2024-11-26 16:48:09 +01:00
parent 0441e19bc9
commit 4035302dd4
3 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/tests/test_pgvector.py
View file

@ -3,6 +3,7 @@ import logging
import pathlib import pathlib
import cognee import cognee
from cognee.api.v1.search import SearchType from cognee.api.v1.search import SearchType
from cognee.modules.retrieval.brute_force_triplet_search import brute_force_triplet_search
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
@ -89,6 +90,9 @@ async def main():
history = await cognee.get_search_history() history = await cognee.get_search_history()
assert len(history) == 6, "Search history is not correct." assert len(history) == 6, "Search history is not correct."
results = await brute_force_triplet_search('Who has the most experience with graphic design?')
assert len(results) > 0
await cognee.prune.prune_data() await cognee.prune.prune_data()
assert not os.path.isdir(data_directory_path), "Local data files are not deleted" assert not os.path.isdir(data_directory_path), "Local data files are not deleted"

4
cognee/tests/test_qdrant.py
View file

@ -5,6 +5,7 @@ import logging
import pathlib import pathlib
import cognee import cognee
from cognee.api.v1.search import SearchType from cognee.api.v1.search import SearchType
from cognee.modules.retrieval.brute_force_triplet_search import brute_force_triplet_search
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
@ -61,6 +62,9 @@ async def main():
history = await cognee.get_search_history() history = await cognee.get_search_history()
assert len(history) == 6, "Search history is not correct." assert len(history) == 6, "Search history is not correct."
results = await brute_force_triplet_search('Who has the most experience with graphic design?')
assert len(results) > 0
await cognee.prune.prune_data() await cognee.prune.prune_data()
assert not os.path.isdir(data_directory_path), "Local data files are not deleted" assert not os.path.isdir(data_directory_path), "Local data files are not deleted"

4
cognee/tests/test_weaviate.py
View file

@ -3,6 +3,7 @@ import logging
import pathlib import pathlib
import cognee import cognee
from cognee.api.v1.search import SearchType from cognee.api.v1.search import SearchType
from cognee.modules.retrieval.brute_force_triplet_search import brute_force_triplet_search
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
@ -59,6 +60,9 @@ async def main():
history = await cognee.get_search_history() history = await cognee.get_search_history()
assert len(history) == 6, "Search history is not correct." assert len(history) == 6, "Search history is not correct."
results = await brute_force_triplet_search('Who has the most experience with graphic design?')
assert len(results) > 0
await cognee.prune.prune_data() await cognee.prune.prune_data()
assert not os.path.isdir(data_directory_path), "Local data files are not deleted" assert not os.path.isdir(data_directory_path), "Local data files are not deleted"