Added failing test based on issue #53

noamgat · Jan 7, 2024 · c1bb065 · c1bb065
1 parent 1ee9d39
commit c1bb065
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_jsonschemaparser.py b/tests/test_jsonschemaparser.py
@@ -303,3 +303,18 @@ def test_more_string_constraints():
         'min_4_max_6': '12_'
     }.items():
         _test_json_schema_parsing_with_string(f'{{"{k}": "{v}"}}', StringConstraints.model_json_schema(), False)
+
+
+def test_union_typed_arrays():
+    class AppleSchema(BaseModel):
+        apple_type: int
+
+    class BananaSchema(BaseModel):
+        is_ripe: bool
+
+    class FruitsSchema(BaseModel):
+        fruits: List[Union[AppleSchema, BananaSchema]]
+
+    _test_json_schema_parsing_with_string('{"fruits": [{"apple_type": 1}, {"apple_type": 2}] }', FruitsSchema.model_json_schema(), True)
+    _test_json_schema_parsing_with_string('{"fruits": [{"apple_type": 1}, {"is_ripe": true}] }', FruitsSchema.model_json_schema(), True)
+    _test_json_schema_parsing_with_string('{"fruits": [{"apple_type": 1, "is_ripe": true}] }', FruitsSchema.model_json_schema(), False)