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

Rename ground_truth to expected_chunks

Browse source
This commit is contained in:
Leon Luithlen 2024-11-13 14:41:28 +01:00
parent cd80525420
commit 49bc07d30d
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/tests/unit/processing/chunks/chunk_by_paragraph_test.py
View file

@ -47,18 +47,18 @@ INPUT_TEXT = {
} }
def run_chunking_test(test_text, ground_truth): def run_chunking_test(test_text, expected_chunks):
chunks = [] chunks = []
for chunk_data in chunk_by_paragraph(test_text, 12, batch_paragraphs=False): for chunk_data in chunk_by_paragraph(test_text, 12, batch_paragraphs=False):
chunks.append(chunk_data) chunks.append(chunk_data)
assert len(chunks) == 3 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"]: for key in ["text", "word_count", "cut_type"]:
assert ( assert (
chunk[key] == ground_truth_item[key] chunk[key] == expected_chunks_item[key]
), f"{key = }: {chunk[key] = } != {ground_truth_item[key] = }" ), f"{key = }: {chunk[key] = } != {expected_chunks_item[key] = }"
def test_chunking_whole_text(): def test_chunking_whole_text():