Initial MVP

2026-03-23 20:38:42 +00:00
commit f245c24928
57 changed files with 6812 additions and 0 deletions
@@ -0,0 +1,120 @@
+"""Tests for output formatters."""
+
+import json
+
+from kb_search.output import (
+    _human_size,
+    format_doc_info,
+    format_document_list,
+    format_search_results,
+    format_status,
+    format_tags,
+)
+
+SAMPLE_SEARCH = {
+    "query": "install git",
+    "results": [
+        {
+            "chunk_id": 1,
+            "score": 0.031,
+            "score_breakdown": {"fts": 0.016, "vector": 0.015},
+            "text": "To install git from source...",
+            "source": {
+                "document_id": 42,
+                "title": "Git Admin Guide",
+                "path": "/docs/git.pdf",
+                "type": "pdf",
+                "page": 12,
+                "section_header": None,
+                "chunk_index": 3,
+                "total_chunks": 28,
+                "tags": ["git", "admin"],
+            },
+        }
+    ],
+    "total_matches": 47,
+    "returned": 1,
+}
+
+
+class TestSearchOutput:
+    def test_json_format(self):
+        output = format_search_results(SAMPLE_SEARCH, "json")
+        parsed = json.loads(output)
+        assert parsed["query"] == "install git"
+        assert len(parsed["results"]) == 1
+        assert parsed["results"][0]["chunk_id"] == 1
+        assert "fts" in parsed["results"][0]["score_breakdown"]
+        assert "vector" in parsed["results"][0]["score_breakdown"]
+
+    def test_json_schema_fields(self):
+        output = format_search_results(SAMPLE_SEARCH, "json")
+        parsed = json.loads(output)
+        r = parsed["results"][0]
+        assert "chunk_id" in r
+        assert "score" in r
+        assert "text" in r
+        assert "source" in r
+        src = r["source"]
+        assert "document_id" in src
+        assert "title" in src
+        assert "type" in src
+        assert "tags" in src
+
+    def test_human_format(self):
+        output = format_search_results(SAMPLE_SEARCH, "human")
+        assert "install git" in output
+        assert "Git Admin Guide" in output
+        assert "p.12" in output
+        assert "0.031" in output
+
+
+class TestDocList:
+    def test_json(self):
+        docs = [{"id": 1, "title": "Test", "type": "pdf", "tags": ["a"], "chunk_count": 5, "created_at": "2024-01-01"}]
+        parsed = json.loads(format_document_list(docs, "json"))
+        assert len(parsed) == 1
+
+    def test_human_empty(self):
+        assert "No documents" in format_document_list([], "human")
+
+    def test_human(self):
+        docs = [{"id": 1, "title": "Test", "type": "pdf", "tags": ["a"], "chunk_count": 5}]
+        output = format_document_list(docs, "human")
+        assert "Test" in output
+
+
+class TestTags:
+    def test_json(self):
+        tags = [{"name": "git", "count": 15}]
+        parsed = json.loads(format_tags(tags, "json"))
+        assert parsed[0]["name"] == "git"
+
+    def test_human_empty(self):
+        assert "No tags" in format_tags([], "human")
+
+
+class TestStatus:
+    def test_json(self):
+        status = {"model_name": "test", "embedding_dim": 384, "schema_version": 1,
+                  "db_size_bytes": 1024, "documents": {"pdf": 5}, "total_documents": 5, "total_chunks": 50}
+        parsed = json.loads(format_status(status, "json"))
+        assert parsed["model_name"] == "test"
+
+    def test_human(self):
+        status = {"model_name": "test", "embedding_dim": 384, "schema_version": 1,
+                  "db_size_bytes": 1024000, "documents": {"pdf": 5}, "total_documents": 5, "total_chunks": 50}
+        output = format_status(status, "human")
+        assert "test" in output
+        assert "384" in output
+
+
+class TestHumanSize:
+    def test_bytes(self):
+        assert _human_size(512) == "512.0 B"
+
+    def test_kb(self):
+        assert _human_size(2048) == "2.0 KB"
+
+    def test_mb(self):
+        assert _human_size(5 * 1024 * 1024) == "5.0 MB"