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:
Daniil Fajnberg
2023-03-10 18:01:20 +01:00
parent bff3c7ef52
commit 72218fd2cb
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)