blob: cf963cf00a1b5e0a32135a24ab8e4cf0fc987982 (
plain) (
blame)
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
|
from pyrsistent._compat import Enum
from pyrsistent import field, pvector_field
# NB: This derives from the internal `pyrsistent._compat.Enum` in order to
# simplify coverage across python versions. Since we use
# `pyrsistent._compat.Enum` in `pyrsistent`'s implementation, it's useful to
# use it in the test coverage as well, for consistency.
class TestEnum(Enum):
x = 1
y = 2
def test_enum():
f = field(type=TestEnum)
assert TestEnum in f.type
assert len(f.type) == 1
# This is meant to exercise `_seq_field`.
def test_pvector_field_enum_type():
f = pvector_field(TestEnum)
assert len(f.type) == 1
assert TestEnum is list(f.type)[0].__type__
|
