#

Author: AlexPetul

tests/test_openapi_schema.py

@@ -93,15 +93,13 @@ def test_map_serializer_field_const(schema: dict, expected: Any):
     json_field = CustomJSONField(schema=schema)
     extension = CustomJSONFieldSerializerExtension(json_field)
     data = extension.map_serializer_field(Mock(), "response")
-    assert data == schema
+    assert data == expected
 
 
 @pytest.mark.parametrize(
     "schema",
     [
-        {
-            "type": "unknown_type",
-        },
+        {"type": "unknown_type"},
     ],
 )
 @patch(
@@ -114,4 +112,4 @@ def test_map_serializer_field_fallback(schema: Any):
     json_field = CustomJSONField(schema=schema)
     extension = CustomJSONFieldSerializerExtension(json_field)
     data = extension.map_serializer_field(Mock(), "response")
-    assert data == {"type": "string"}
+    assert data == {"type": "string"}