fix test

langchain-ai · Nov 29, 2024 · 5b031da · 5b031da
1 parent 9dfaf99
commit 5b031da
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libs/upstage/tests/integration_tests/test_document_parse.py b/libs/upstage/tests/integration_tests/test_document_parse.py
@@ -47,7 +47,9 @@ def test_document_parse(
         base64_encoding=[base64_encoding],
     )
     documents = loader.load()
-    if split == "element":
+    if split == "element" and ocr == "auto":
         assert len(documents) == 14
+    if split == "element" and ocr == "force":
+        assert len(documents) == 15
     else:
         assert len(documents) == 1