From 14dd60576ed080507711d94ea872bb158a7e86ed Mon Sep 17 00:00:00 2001 From: Leon Luithlen Date: Thu, 14 Nov 2024 17:06:16 +0100 Subject: [PATCH] Fix indexing in tests in chunk_by_sentence_test --- cognee/tests/unit/processing/chunks/chunk_by_sentence_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/tests/unit/processing/chunks/chunk_by_sentence_test.py b/cognee/tests/unit/processing/chunks/chunk_by_sentence_test.py index 2f42f836a..d1c75d7ed 100644 --- a/cognee/tests/unit/processing/chunks/chunk_by_sentence_test.py +++ b/cognee/tests/unit/processing/chunks/chunk_by_sentence_test.py @@ -15,7 +15,7 @@ maximum_length_vals = [None, 8, 64] ) def test_chunk_by_sentence_isomorphism(input_text, maximum_length): chunks = chunk_by_sentence(input_text, maximum_length) - reconstructed_text = "".join([chunk[2] for chunk in chunks]) + reconstructed_text = "".join([chunk[1] for chunk in chunks]) assert ( reconstructed_text == input_text ), f"texts are not identical: {len(input_text) = }, {len(reconstructed_text) = }" @@ -33,7 +33,7 @@ def test_chunk_by_sentence_isomorphism(input_text, maximum_length): def test_paragraph_chunk_length(input_text, maximum_length): chunks = list(chunk_by_sentence(input_text, maximum_length)) - chunk_lengths = np.array([len(list(chunk_by_word(chunk[2]))) for chunk in chunks]) + chunk_lengths = np.array([len(list(chunk_by_word(chunk[1]))) for chunk in chunks]) larger_chunks = chunk_lengths[chunk_lengths > maximum_length] assert np.all(