Add "in" model

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

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/pydantic_jsonlogic/__init__.py b/pydantic_jsonlogic/__init__.py
@@ -127,3 +127,9 @@ class Modulo(BaseModel):
     model_config = ConfigDict(strict=True)
 
     modulo: tuple[int, int] = Field(validation_alias="%")
+
+
+class In(BaseModel):
+    model_config = ConfigDict(strict=True)
+
+    in_: tuple[str, list[str] | str] = Field(validation_alias="in")
diff --git a/tests/test_fields.py b/tests/test_fields.py
@@ -6,6 +6,7 @@
     Equals,
     GreaterThan,
     GreaterThanOrEqual,
+    In,
     LessThan,
     LessThanOrEqual,
     Max,
@@ -272,3 +273,14 @@ def test_divide(json: str) -> None:
 )
 def test_modulo(json: str) -> None:
     Modulo.model_validate_json(json)
+
+
+@pytest.mark.parametrize(
+    "json",
+    [
+        '{"in":["Bart",["Bart","Homer","Lisa","Marge","Maggie"]]}',
+        '{"in":["i","team"]}',
+    ],
+)
+def test_in(json: str) -> None:
+    In.model_validate_json(json)