test: Fix test for multimedia deduplication

Add missing function to get data from database to multimedia deduplication test Test COG-505
topoteretes · Dec 5, 2024 · 9ba5d49 · 9ba5d49
1 parent add6730
commit 9ba5d49
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cognee/tests/test_deduplication.py b/cognee/tests/test_deduplication.py
@@ -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()