mirror of
https://github.com/kennethreitz/pydantic.git
synced 2026-06-05 23:00:18 +00:00
Samuel Colvin
1161df3ca0
* support ForwardRef on dataclasses, fix #397 * explicit docs about when postponed annotations don't work
143 lines
2.9 KiB
Python
143 lines
2.9 KiB
Python
"""
|
|
Tests for python 3.7 behaviour, eg postponed annotations and ForwardRef.
|
|
"""
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from pydantic import ConfigError
|
|
|
|
skip_not_37 = pytest.mark.skipif(sys.version_info < (3, 7), reason='testing >= 3.7 behaviour only')
|
|
|
|
|
|
@skip_not_37
|
|
def test_postponed_annotations(create_module):
|
|
module = create_module(
|
|
"""
|
|
from __future__ import annotations
|
|
from pydantic import BaseModel
|
|
|
|
class Model(BaseModel):
|
|
a: int
|
|
"""
|
|
)
|
|
m = module.Model(a='123')
|
|
assert m.dict() == {'a': 123}
|
|
|
|
|
|
@skip_not_37
|
|
def test_postponed_annotations_optional(create_module):
|
|
module = create_module(
|
|
"""
|
|
from __future__ import annotations
|
|
from typing import Optional
|
|
from pydantic import BaseModel
|
|
|
|
class Model(BaseModel):
|
|
a: Optional[int]
|
|
"""
|
|
)
|
|
assert module.Model(a='123').dict() == {'a': 123}
|
|
assert module.Model().dict() == {'a': None}
|
|
|
|
|
|
@skip_not_37
|
|
def test_basic_forward_ref(create_module):
|
|
module = create_module(
|
|
"""
|
|
from typing import ForwardRef, Optional
|
|
from pydantic import BaseModel
|
|
|
|
class Foo(BaseModel):
|
|
a: int
|
|
|
|
FooRef = ForwardRef('Foo')
|
|
|
|
class Bar(BaseModel):
|
|
b: Optional[FooRef]
|
|
"""
|
|
)
|
|
|
|
assert module.Bar().dict() == {'b': None}
|
|
assert module.Bar(b={'a': '123'}).dict() == {'b': {'a': 123}}
|
|
|
|
|
|
@skip_not_37
|
|
def test_self_forward_ref_module(create_module):
|
|
module = create_module(
|
|
"""
|
|
from typing import ForwardRef
|
|
from pydantic import BaseModel
|
|
|
|
Foo = ForwardRef('Foo')
|
|
|
|
class Foo(BaseModel):
|
|
a: int = 123
|
|
b: Foo = None
|
|
|
|
Foo.update_forward_refs()
|
|
"""
|
|
)
|
|
|
|
assert module.Foo().dict() == {'a': 123, 'b': None}
|
|
assert module.Foo(b={'a': '321'}).dict() == {'a': 123, 'b': {'a': 321, 'b': None}}
|
|
|
|
|
|
@skip_not_37
|
|
def test_self_forward_ref_local(create_module):
|
|
module = create_module(
|
|
"""
|
|
from typing import ForwardRef
|
|
from pydantic import BaseModel
|
|
|
|
def main():
|
|
Foo = ForwardRef('Foo')
|
|
|
|
class Foo(BaseModel):
|
|
a: int = 123
|
|
b: Foo = None
|
|
|
|
Foo.update_forward_refs()
|
|
return Foo
|
|
"""
|
|
)
|
|
Foo = module.main()
|
|
assert Foo().dict() == {'a': 123, 'b': None}
|
|
assert Foo(b={'a': '321'}).dict() == {'a': 123, 'b': {'a': 321, 'b': None}}
|
|
|
|
|
|
@skip_not_37
|
|
def test_missing_update_forward_refs(create_module):
|
|
module = create_module(
|
|
"""
|
|
from typing import ForwardRef
|
|
from pydantic import BaseModel
|
|
|
|
Foo = ForwardRef('Foo')
|
|
|
|
class Foo(BaseModel):
|
|
a: int = 123
|
|
b: Foo = None
|
|
"""
|
|
)
|
|
with pytest.raises(ConfigError) as exc_info:
|
|
module.Foo(b=123)
|
|
assert str(exc_info.value).startswith('field "b" not yet prepared so type is still a ForwardRef')
|
|
|
|
|
|
@skip_not_37
|
|
def test_forward_ref_dataclass(create_module):
|
|
module = create_module(
|
|
"""
|
|
from __future__ import annotations
|
|
from pydantic import UrlStr
|
|
from pydantic.dataclasses import dataclass
|
|
|
|
@dataclass
|
|
class Dataclass:
|
|
url: UrlStr
|
|
"""
|
|
)
|
|
m = module.Dataclass('http://example.com ')
|
|
assert m.url == 'http://example.com'
|