daniil-berg/marshmallow-generic
1
0
Fork 0
generated from daniil-berg/boilerplate-py
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✨ Add loads method overloads to GenericSchema class

Browse Source
This commit is contained in:
Daniel Fainberg
2023-03-10 18:01:20 +01:00
parent 70ce0ee2b3
commit 0b71633ae5
2 changed files with 51 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_schema.py
View File

@@ -20,7 +20,7 @@ class GenericSchemaTestCase(TestCase):
mock__get_type_arg.assert_called_once_with()
mock_cls.assert_called_once_with(**mock_data)
def test_load(self) -> None:
def test_load_and_loads(self) -> None:
"""Mainly for static type checking purposes."""
class Foo:
@@ -29,9 +29,16 @@ class GenericSchemaTestCase(TestCase):
class TestSchema(schema.GenericSchema[Foo]):
pass
single: Foo = TestSchema().load({})
single: Foo
single = TestSchema().load({})
self.assertIsInstance(single, Foo)
single = TestSchema().loads("{}")
self.assertIsInstance(single, Foo)
multiple: list[Foo] = TestSchema().load([{}], many=True)
multiple: list[Foo]
multiple = TestSchema().load([{}], many=True)
self.assertIsInstance(multiple, list)
self.assertIsInstance(multiple[0], Foo)
multiple = TestSchema().loads("[{}]", many=True)
self.assertIsInstance(multiple, list)
self.assertIsInstance(multiple[0], Foo)