core[patch]: Add unit test for str and repr for Document (#23414)

langchain-ai · Jun 25, 2024 · 7e9e69c · 7e9e69c
1 parent f055f2a
commit 7e9e69c
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/libs/core/tests/unit_tests/documents/test_str.py b/libs/core/tests/unit_tests/documents/test_str.py
@@ -0,0 +1,20 @@
+from langchain_core.documents import Document
+
+
+def test_str() -> None:
+    assert str(Document(page_content="Hello, World!")) == "page_content='Hello, World!'"
+    assert (
+        str(Document(page_content="Hello, World!", metadata={"a": 3}))
+        == "page_content='Hello, World!' metadata={'a': 3}"
+    )
+
+
+def test_repr() -> None:
+    assert (
+        repr(Document(page_content="Hello, World!"))
+        == "Document(page_content='Hello, World!')"
+    )
+    assert (
+        repr(Document(page_content="Hello, World!", metadata={"a": 3}))
+        == "Document(page_content='Hello, World!', metadata={'a': 3})"
+    )