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

test: Fix test for multimedia deduplication

Browse source 
Add missing function to get data from database to multimedia deduplication test

Test COG-505
This commit is contained in:
Igor Ilic 2024-12-05 20:09:29 +01:00
parent add6730b9e
commit 9ba5d49e69
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/tests/test_deduplication.py
View file

@ -74,6 +74,7 @@ async def test_deduplication():
await cognee.add([explanation_file_path], dataset_name)
await cognee.add([explanation_file_path2], dataset_name2)
result = await relational_engine.get_all_data_from_table("data")
assert len(result) == 1, "More than one data entity was found."
await cognee.prune.prune_data()
@ -90,6 +91,7 @@ async def test_deduplication():
await cognee.add([explanation_file_path], dataset_name)
await cognee.add([explanation_file_path2], dataset_name2)
result = await relational_engine.get_all_data_from_table("data")
assert len(result) == 1, "More than one data entity was found."
await cognee.prune.prune_data()