From 49bc07d30d6072565bf5b84c5cd0c512fc693f1f Mon Sep 17 00:00:00 2001 From: Leon Luithlen Date: Wed, 13 Nov 2024 14:41:28 +0100 Subject: [PATCH] Rename ground_truth to expected_chunks --- .../unit/processing/chunks/chunk_by_paragraph_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cognee/tests/unit/processing/chunks/chunk_by_paragraph_test.py b/cognee/tests/unit/processing/chunks/chunk_by_paragraph_test.py index fb35ef7fb..24c3cc147 100644 --- a/cognee/tests/unit/processing/chunks/chunk_by_paragraph_test.py +++ b/cognee/tests/unit/processing/chunks/chunk_by_paragraph_test.py @@ -47,18 +47,18 @@ INPUT_TEXT = { } -def run_chunking_test(test_text, ground_truth): +def run_chunking_test(test_text, expected_chunks): chunks = [] for chunk_data in chunk_by_paragraph(test_text, 12, batch_paragraphs=False): chunks.append(chunk_data) assert len(chunks) == 3 - for ground_truth_item, chunk in zip(ground_truth, chunks): + for expected_chunks_item, chunk in zip(expected_chunks, chunks): for key in ["text", "word_count", "cut_type"]: assert ( - chunk[key] == ground_truth_item[key] - ), f"{key = }: {chunk[key] = } != {ground_truth_item[key] = }" + chunk[key] == expected_chunks_item[key] + ), f"{key = }: {chunk[key] = } != {expected_chunks_item[key] = }" def test_chunking_whole_text():