added unit tests

Author: epinzur

libs/knowledge-store/tests/unit_tests/test_metadata_policy_normalization.py

@@ -0,0 +1,30 @@
+"""
+Normalization of metadata policy specification options
+"""
+
+from ragstack_knowledge_store.graph_store import MetadataIndexingMode, GraphStore
+
+
+class TestNormalizeMetadataPolicy:
+    def test_normalize_metadata_policy(self) -> None:
+        #
+        mdp1 = GraphStore._normalize_metadata_indexing_policy("all")
+        assert mdp1 == (MetadataIndexingMode.DEFAULT_TO_SEARCHABLE, set())
+        #
+        mdp2 = GraphStore._normalize_metadata_indexing_policy("none")
+        assert mdp2 == (MetadataIndexingMode.DEFAULT_TO_UNSEARCHABLE, set())
+        #
+        mdp3 = GraphStore._normalize_metadata_indexing_policy(
+            ("default_to_Unsearchable", ["x", "y"]),
+        )
+        assert mdp3 == (MetadataIndexingMode.DEFAULT_TO_UNSEARCHABLE, {"x", "y"})
+        #
+        mdp4 = GraphStore._normalize_metadata_indexing_policy(
+            ("DenyList", ["z"]),
+        )
+        assert mdp4 == (MetadataIndexingMode.DEFAULT_TO_SEARCHABLE, {"z"})
+        # s
+        mdp5 = GraphStore._normalize_metadata_indexing_policy(
+            ("deny_LIST", "singlefield")
+        )
+        assert mdp5 == (MetadataIndexingMode.DEFAULT_TO_SEARCHABLE, {"singlefield"})

libs/knowledge-store/tests/unit_tests/test_metadata_string_coercion.py

@@ -0,0 +1,30 @@
+"""
+Stringification of everything in the simple metadata handling
+"""
+
+from ragstack_knowledge_store.graph_store import GraphStore, SetupMode
+
+
+class TestMetadataStringCoercion:
+    def test_metadata_string_coercion(self) -> None:
+        md_dict = {
+                "integer": 1,
+                "float": 2.0,
+                "boolean": True,
+                "null": None,
+                "string": "letter E",
+                "something": RuntimeError("You cannot do this!"),
+            }
+
+        stringified = {k: GraphStore._coerce_string(v) for k, v in md_dict.items()}
+
+        expected = {
+            "integer": "1.0",
+            "float": "2.0",
+            "boolean": "true",
+            "null": "null",
+            "string": "letter E",
+            "something": str(RuntimeError("You cannot do this!")),
+        }
+
+        assert stringified == expected