fix tests and coverage for pydantic#1209

samuelcolvin · samuelcolvin · commit 5067508eca62 · 2020-05-01T10:32:39.000+01:00
diff --git a/tests/test_schema.py b/tests/test_schema.py
@@ -967,13 +967,16 @@ class Foo(BaseModel):
     class Bar(BaseModel):
         pass
 
-    class Model_1(BaseModel):
-        a: Union[Foo, Bar]
-
-    class Model_2(BaseModel):
-        a: Union[Foo, Bar] = Field(...)
-
-    assert Model_1.schema()['properties'] == Model_2.schema()['properties']
+    class Spam(BaseModel):
+        a: Union[Foo, Bar] = Field(..., description='xxx')
+
+    assert Spam.schema()['properties'] == {
+        'a': {
+            'title': 'A',
+            'description': 'xxx',
+            'anyOf': [{'$ref': '#/definitions/Foo'}, {'$ref': '#/definitions/Bar'}],
+        },
+    }
 
 
 def test_schema_from_models():
