mirror of
https://github.com/kennethreitz/pydantic.git
synced 2026-06-05 23:00:18 +00:00
Samuel Colvin
37364a05b9
* implement JSON serialisation, fix #133 * documenting JSON serialisation * fix coverage
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
import datetime
|
|
import json
|
|
from decimal import Decimal
|
|
from enum import Enum
|
|
from uuid import UUID
|
|
|
|
import pytest
|
|
|
|
from pydantic import BaseModel, create_model
|
|
from pydantic.json import pydantic_encoder
|
|
|
|
|
|
class MyEnum(Enum):
|
|
foo = 'bar'
|
|
snap = 'crackle'
|
|
|
|
|
|
@pytest.mark.parametrize('input,output', [
|
|
(UUID('ebcdab58-6eb8-46fb-a190-d07a33e9eac8'), '"ebcdab58-6eb8-46fb-a190-d07a33e9eac8"'),
|
|
(datetime.datetime(2032, 1, 1, 1, 1), '"2032-01-01T01:01:00"'),
|
|
(datetime.datetime(2032, 1, 1, 1, 1, tzinfo=datetime.timezone.utc), '"2032-01-01T01:01:00+00:00"'),
|
|
(datetime.datetime(2032, 1, 1), '"2032-01-01T00:00:00"'),
|
|
(datetime.time(12, 34, 56), '"12:34:56"'),
|
|
({1, 2, 3}, '[1, 2, 3]'),
|
|
(frozenset([1, 2, 3]), '[1, 2, 3]'),
|
|
((v for v in range(4)), '[0, 1, 2, 3]'),
|
|
(b'this is bytes', '"this is bytes"'),
|
|
(Decimal('12.34'), '12.34'),
|
|
(create_model('BarModel', a='b', c='d')(), '{"a": "b", "c": "d"}'),
|
|
(MyEnum.foo, '"bar"')
|
|
])
|
|
def test_encoding(input, output):
|
|
assert json.dumps(input, default=pydantic_encoder) == output
|
|
|
|
|
|
def test_model_encoding():
|
|
class ModelA(BaseModel):
|
|
x: int
|
|
y: str
|
|
|
|
class Model(BaseModel):
|
|
a: float
|
|
b: bytes
|
|
c: Decimal
|
|
d: ModelA
|
|
|
|
m = Model(a=10.2, b='foobar', c=10.2, d={'x': 123, 'y': '123'})
|
|
assert m.dict() == {'a': 10.2, 'b': b'foobar', 'c': Decimal('10.2'), 'd': {'x': 123, 'y': '123'}}
|
|
assert m.json() == '{"a": 10.2, "b": "foobar", "c": 10.2, "d": {"x": 123, "y": "123"}}'
|
|
assert m.json(exclude={'b'}) == '{"a": 10.2, "c": 10.2, "d": {"x": 123, "y": "123"}}'
|
|
|
|
|
|
def test_invalid_model():
|
|
class Foo:
|
|
pass
|
|
|
|
with pytest.raises(TypeError):
|
|
json.dumps(Foo, default=pydantic_encoder)
|