58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
|
"""
|
||
|
|
Tests for PEP-526 type annotations.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import absolute_import, division, print_function
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from attr._make import (
|
||
|
|
attr,
|
||
|
|
attributes,
|
||
|
|
fields
|
||
|
|
)
|
||
|
|
|
||
|
|
import typing
|
||
|
|
|
||
|
|
|
||
|
|
class TestAnnotations(object):
|
||
|
|
"""
|
||
|
|
Tests for types derived from variable annotations (PEP-526).
|
||
|
|
"""
|
||
|
|
|
||
|
|
def test_basic_annotations(self):
|
||
|
|
"""
|
||
|
|
Sets the `Attribute.type` attr from basic type annotations.
|
||
|
|
"""
|
||
|
|
@attributes
|
||
|
|
class C(object):
|
||
|
|
x: int = attr()
|
||
|
|
y = attr(type=str)
|
||
|
|
z = attr()
|
||
|
|
assert int is fields(C).x.type
|
||
|
|
assert str is fields(C).y.type
|
||
|
|
assert None is fields(C).z.type
|
||
|
|
|
||
|
|
def test_catches_basic_type_conflict(self):
|
||
|
|
"""
|
||
|
|
Raises ValueError type is specified both ways.
|
||
|
|
"""
|
||
|
|
with pytest.raises(ValueError) as e:
|
||
|
|
@attributes
|
||
|
|
class C:
|
||
|
|
x: int = attr(type=int)
|
||
|
|
assert ("Type annotation and type argument cannot "
|
||
|
|
"both be present",) == e.value.args
|
||
|
|
|
||
|
|
def test_typing_annotations(self):
|
||
|
|
"""
|
||
|
|
Sets the `Attribute.type` attr from typing annotations.
|
||
|
|
"""
|
||
|
|
@attributes
|
||
|
|
class C(object):
|
||
|
|
x: typing.List[int] = attr()
|
||
|
|
y = attr(type=typing.Optional[str])
|
||
|
|
|
||
|
|
assert typing.List[int] is fields(C).x.type
|
||
|
|
assert typing.Optional[str] is fields(C).y.type
|