Add "cat" model

Amsterdam · Aug 27, 2024 · d49c091 · d49c091
1 parent ce88ee3
commit d49c091
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/pydantic_jsonlogic/__init__.py b/pydantic_jsonlogic/__init__.py
@@ -133,3 +133,9 @@ class In(BaseModel):
     model_config = ConfigDict(strict=True)
 
     in_: tuple[str, list[str] | str] = Field(validation_alias="in")
+
+
+class Cat(BaseModel):
+    model_config = ConfigDict(strict=True)
+
+    cat: int | str | list[int | str]
diff --git a/tests/test_fields.py b/tests/test_fields.py
@@ -2,6 +2,7 @@
 
 from pydantic_jsonlogic import (
     Add,
+    Cat,
     Divide,
     Equals,
     GreaterThan,
@@ -284,3 +285,18 @@ def test_modulo(json: str) -> None:
 )
 def test_in(json: str) -> None:
     In.model_validate_json(json)
+
+
+@pytest.mark.parametrize(
+    "json",
+    [
+        '{"cat":"ice"}',
+        '{"cat":["ice"]}',
+        '{"cat":["ice","cream"]}',
+        '{"cat":[1,2]}',
+        '{"cat":["Robocop",2]}',
+        '{"cat":["we all scream for ","ice","cream"]}',
+    ],
+)
+def test_cat(json: str) -> None:
+    Cat.model_validate_json(json)