Add test_set_dict_class

marshmallow-code · Jul 16, 2023 · 6abbfca · 6abbfca
1 parent 684fe36
commit 6abbfca
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/test_schema.py b/tests/test_schema.py
@@ -2933,3 +2933,16 @@ class Meta:
             MySchema(unknown="badval")
         else:
             MySchema().load({"foo": "bar"}, unknown="badval")
+
+
+@pytest.mark.parametrize("dict_cls", (dict, OrderedDict))
+def test_set_dict_class(dict_cls):
+    """Demonstrate how to specify dict_class as class attribute"""
+
+    class MySchema(Schema):
+        dict_class = dict_cls
+        foo = fields.String()
+
+    result = MySchema().dump({"foo": "bar"})
+    assert result == {"foo": "bar"}
+    assert isinstance(result, dict_cls)