blob: 176b64cc6bec983adbf6204c8d752c07c1cb5523 (
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
|
from enum import Enum
from pyrsistent import field, pvector_field
class ExampleEnum(Enum):
x = 1
y = 2
def test_enum():
f = field(type=ExampleEnum)
assert ExampleEnum in f.type
assert len(f.type) == 1
# This is meant to exercise `_seq_field`.
def test_pvector_field_enum_type():
f = pvector_field(ExampleEnum)
assert len(f.type) == 1
assert ExampleEnum is list(f.type)[0].__type__
|