From 555082713a0738ddc5f48f6dba9040851b6dee39 Mon Sep 17 00:00:00 2001 From: Umar Butler Date: Sun, 5 Nov 2023 21:33:50 +1100 Subject: [PATCH] Fixed unit test. --- tests/test_semchunk.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_semchunk.py b/tests/test_semchunk.py index f39f5cc..d90213e 100644 --- a/tests/test_semchunk.py +++ b/tests/test_semchunk.py @@ -20,7 +20,7 @@ def test_chunk(): # Test a variety of chunk sizes. for chunk_size in range(1,100): for chunk in semchunk.chunk(LOREM, chunk_size=chunk_size, token_counter=_token_counter): - assert _token_counter(chunk) <= 1 + assert _token_counter(chunk) <= chunk_size # Test a chunk size larger than the text. semchunk.chunk(LOREM, chunk_size=len(LOREM)**2, token_counter=_token_counter) \ No newline at end of file