From 9ba5d49e696a66eac0f384c8fed4d09068a68a0c Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 5 Dec 2024 20:09:29 +0100 Subject: [PATCH] test: Fix test for multimedia deduplication Add missing function to get data from database to multimedia deduplication test Test COG-505 --- cognee/tests/test_deduplication.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/cognee/tests/test_deduplication.py b/cognee/tests/test_deduplication.py index 79936cba5..467a52368 100644 --- 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()