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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
import pickle
import pytest
from pyrsistent import CheckedPMap, InvariantException, PMap, CheckedType, CheckedPSet, CheckedPVector, \
CheckedKeyTypeError, CheckedValueTypeError
class FloatToIntMap(CheckedPMap):
__key_type__ = float
__value_type__ = int
__invariant__ = lambda key, value: (int(key) == value, 'Invalid mapping')
def test_instantiate():
x = FloatToIntMap({1.25: 1, 2.5: 2})
assert dict(x.items()) == {1.25: 1, 2.5: 2}
assert isinstance(x, FloatToIntMap)
assert isinstance(x, PMap)
assert isinstance(x, CheckedType)
def test_instantiate_empty():
x = FloatToIntMap()
assert dict(x.items()) == {}
assert isinstance(x, FloatToIntMap)
def test_set():
x = FloatToIntMap()
x2 = x.set(1.0, 1)
assert x2[1.0] == 1
assert isinstance(x2, FloatToIntMap)
def test_invalid_key_type():
with pytest.raises(CheckedKeyTypeError):
FloatToIntMap({1: 1})
def test_invalid_value_type():
with pytest.raises(CheckedValueTypeError):
FloatToIntMap({1.0: 1.0})
def test_breaking_invariant():
try:
FloatToIntMap({1.5: 2})
assert False
except InvariantException as e:
assert e.invariant_errors == ('Invalid mapping',)
def test_repr():
x = FloatToIntMap({1.25: 1})
assert str(x) == 'FloatToIntMap({1.25: 1})'
def test_default_serialization():
x = FloatToIntMap({1.25: 1, 2.5: 2})
assert x.serialize() == {1.25: 1, 2.5: 2}
class StringFloatToIntMap(FloatToIntMap):
@staticmethod
def __serializer__(format, key, value):
return format.format(key), format.format(value)
def test_custom_serialization():
x = StringFloatToIntMap({1.25: 1, 2.5: 2})
assert x.serialize("{0}") == {"1.25": "1", "2.5": "2"}
class FloatSet(CheckedPSet):
__type__ = float
class IntToFloatSetMap(CheckedPMap):
__key_type__ = int
__value_type__ = FloatSet
def test_multi_level_serialization():
x = IntToFloatSetMap.create({1: [1.25, 1.50], 2: [2.5, 2.75]})
assert str(x) == "IntToFloatSetMap({1: FloatSet([1.5, 1.25]), 2: FloatSet([2.75, 2.5])})"
sx = x.serialize()
assert sx == {1: set([1.5, 1.25]), 2: set([2.75, 2.5])}
assert isinstance(sx[1], set)
def test_create_non_checked_types():
assert FloatToIntMap.create({1.25: 1, 2.5: 2}) == FloatToIntMap({1.25: 1, 2.5: 2})
def test_create_checked_types():
class IntSet(CheckedPSet):
__type__ = int
class FloatVector(CheckedPVector):
__type__ = float
class IntSetToFloatVectorMap(CheckedPMap):
__key_type__ = IntSet
__value_type__ = FloatVector
x = IntSetToFloatVectorMap.create({frozenset([1, 2]): [1.25, 2.5]})
assert str(x) == "IntSetToFloatVectorMap({IntSet([1, 2]): FloatVector([1.25, 2.5])})"
def test_evolver_returns_same_instance_when_no_updates():
x = FloatToIntMap({1.25: 1, 2.25: 2})
assert x.evolver().persistent() is x
def test_map_with_no_types_or_invariants():
class NoCheckPMap(CheckedPMap):
pass
x = NoCheckPMap({1: 2, 3: 4})
assert x[1] == 2
assert x[3] == 4
def test_pickling():
x = FloatToIntMap({1.25: 1, 2.5: 2})
y = pickle.loads(pickle.dumps(x, -1))
assert x == y
assert isinstance(y, FloatToIntMap)
class FloatVector(CheckedPVector):
__type__ = float
class VectorToSetMap(CheckedPMap):
__key_type__ = 'checked_map_test.FloatVector'
__value_type__ = 'checked_map_test.FloatSet'
def test_type_check_with_string_specification():
content = [1.5, 2.0]
vec = FloatVector(content)
sett = FloatSet(content)
map = VectorToSetMap({vec: sett})
assert map[vec] == sett
def test_type_creation_with_string_specification():
content = (1.5, 2.0)
map = VectorToSetMap.create({content: content})
assert map[FloatVector(content)] == set(content)
def test_supports_weakref():
import weakref
weakref.ref(VectorToSetMap({}))
|
