Fixed unit test.

umarbutler · Nov 5, 2023 · 5550827 · 5550827
1 parent f5ef6d3
commit 5550827
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)