Fix tests

benbrandt · Dec 27, 2023 · 73152f6 · 73152f6
1 parent d9f460c
commit 73152f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bindings/python/tests/test_integration.py b/bindings/python/tests/test_integration.py
@@ -32,14 +32,14 @@ def test_hugging_face():
     tokenizer = Tokenizer.from_pretrained("bert-base-uncased")
     splitter = HuggingFaceTextSplitter(tokenizer, trim_chunks=False)
     text = "123\n123"
-    assert splitter.chunks(text, 1) == ["123\n", "123"]
+    assert splitter.chunks(text, 1) == ["123", "\n123"]
 
 
 def test_hugging_face_range():
     tokenizer = Tokenizer.from_pretrained("bert-base-uncased")
     splitter = HuggingFaceTextSplitter(tokenizer, trim_chunks=False)
     text = "123\n123"
-    assert splitter.chunks(text=text, chunk_capacity=(1, 2)) == ["123\n", "123"]
+    assert splitter.chunks(text=text, chunk_capacity=(1, 2)) == ["123", "\n123"]
 
 
 def test_hugging_face_trim():