tests: vectors

tests/vectorstore/test_document_class.py

@@ -0,0 +1,39 @@
+import pytest
+from application.vectorstore.document_class import Document
+
+
+@pytest.mark.unit
+class TestDocument:
+    def test_create_document(self):
+        doc = Document(page_content="hello world", metadata={"source": "test"})
+        assert doc.page_content == "hello world"
+        assert doc.metadata == {"source": "test"}
+
+    def test_document_is_string(self):
+        doc = Document(page_content="hello world", metadata={})
+        assert isinstance(doc, str)
+        assert str(doc) == "hello world"
+
+    def test_document_string_equality(self):
+        doc = Document(page_content="hello", metadata={"k": "v"})
+        assert doc == "hello"
+
+    def test_document_empty_metadata(self):
+        doc = Document(page_content="text", metadata={})
+        assert doc.metadata == {}
+
+    def test_document_empty_content(self):
+        doc = Document(page_content="", metadata={"a": 1})
+        assert doc.page_content == ""
+        assert doc == ""
+
+    def test_document_preserves_complex_metadata(self):
+        meta = {"source": "file.txt", "page": 3, "nested": {"key": "val"}}
+        doc = Document(page_content="content", metadata=meta)
+        assert doc.metadata["nested"]["key"] == "val"
+
+    def test_document_string_operations(self):
+        doc = Document(page_content="hello world", metadata={})
+        assert doc.upper() == "HELLO WORLD"
+        assert doc.split() == ["hello", "world"]
+        assert "world" in doc