1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
"""Tests for enums."""
from hypothesis import given
from hypothesis.strategies import data, sampled_from
from pytest import raises
from cattrs import BaseConverter
from cattrs._compat import Literal
from .untyped import enums_of_primitives
@given(data(), enums_of_primitives())
def test_structuring_enums(data, enum):
"""Test structuring enums by their values."""
converter = BaseConverter()
val = data.draw(sampled_from(list(enum)))
assert converter.structure(val.value, enum) == val
@given(enums_of_primitives())
def test_enum_failure(enum):
"""Structuring literals with enums fails properly."""
converter = BaseConverter()
type = Literal[next(iter(enum))]
with raises(Exception) as exc_info:
converter.structure("", type)
assert exc_info.value.args[0] == f" not in literal {type!r}"
|
