Add "map" model

Amsterdam · Aug 28, 2024 · c62ebd3 · c62ebd3
1 parent 312e594
commit c62ebd3
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/pydantic_jsonlogic/__init__.py b/pydantic_jsonlogic/__init__.py
@@ -142,4 +142,10 @@ class Filter(BaseJSONLogicOperation):
     ]
 
 
-class Map(BaseJSONLogicOperation): ...
+class Map(BaseJSONLogicOperation):
+    map: tuple[
+        # An array or an operation that produces an array
+        Union[list[Any], Var, Missing, MissingSome, If, Merge, Filter, "Map"],
+        # The operation to be performed on each element of the array
+        BaseJSONLogicOperation,
+    ]
diff --git a/tests/test_fields.py b/tests/test_fields.py
@@ -14,6 +14,7 @@
     LessThan,
     LessThanOrEqual,
     Log,
+    Map,
     Max,
     Merge,
     Min,
@@ -493,3 +494,15 @@ def test_if(json: str) -> None:
 )
 def test_filter(json: str) -> None:
     Filter.model_validate_json(json)
+
+
+@pytest.mark.parametrize(
+    "json",
+    [
+        '{"map":[{"var":"integers"}, {"*":[{"var":""},2]}]}',
+        '{"map":[{"var":"integers"}, {"*":[{"var":""},2]}]}',
+        '{"map":[{"var":"desserts"}, {"var":"qty"}]}',
+    ],
+)
+def test_map(json: str) -> None:
+    Map.model_validate_json(json)