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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
|
import test_enum_ext as t
import pytest
def test01_unsigned_enum():
assert repr(t.Enum.A) == 'Enum.A'
assert str(t.Enum.A) == 'Enum.A'
assert repr(t.Enum.B) == 'Enum.B'
assert str(t.Enum.B) == 'Enum.B'
assert repr(t.Enum.C) == 'Enum.C'
assert str(t.Enum.C) == 'Enum.C'
assert t.Enum.A.name == 'A'
assert t.Enum.B.name == 'B'
assert t.Enum.C.name == 'C'
assert t.Enum.A.__name__ == 'A'
assert t.Enum.B.__name__ == 'B'
assert t.Enum.C.__name__ == 'C'
assert t.Enum.__doc__ == 'enum-level docstring'
assert t.Enum.A.__doc__ == 'Value A'
assert t.Enum.B.__doc__ == 'Value B'
assert t.Enum.C.__doc__ == 'Value C'
assert t.Enum.A.value == 0
assert t.Enum.B.value == 1
assert t.Enum.C.value == 0xffffffff
assert t.Enum(0) is t.Enum.A
assert t.Enum(1) is t.Enum.B
assert t.Enum(0xffffffff) is t.Enum.C
assert t.Enum(t.Enum.A) is t.Enum.A
assert t.Enum(t.Enum.B) is t.Enum.B
assert t.Enum(t.Enum.C) == t.Enum.C
assert t.from_enum(t.Enum.A) == 0
assert t.from_enum(t.Enum.B) == 1
assert t.from_enum(t.Enum.C) == 0xffffffff
assert t.to_enum(0).__name__ == 'A'
assert t.to_enum(0) == t.Enum.A
assert t.to_enum(1) == t.Enum.B
assert t.to_enum(0xffffffff) == t.Enum.C
with pytest.raises(ValueError) as excinfo:
t.to_enum(5)
assert '5 is not a valid Enum' in str(excinfo.value)
with pytest.raises(ValueError) as excinfo:
t.Enum(123)
assert '123 is not a valid Enum' in str(excinfo.value)
assert t.from_enum_implicit(0) == 0
with pytest.raises(TypeError):
t.from_enum_implicit(123)
def test02_signed_enum():
assert repr(t.SEnum.A) == 'SEnum.A'
assert repr(t.SEnum.B) == 'SEnum.B'
assert repr(t.SEnum.C) == 'SEnum.C'
assert str(t.SEnum.A) == 'SEnum.A'
assert str(t.SEnum.B) == 'SEnum.B'
assert str(t.SEnum.C) == 'SEnum.C'
assert int(t.SEnum.A) == 0
assert int(t.SEnum.B) == 1
assert int(t.SEnum.C) == -1
assert t.SEnum.A.value == 0
assert t.SEnum.B.value == 1
assert t.SEnum.C.value == -1
assert t.SEnum(0) is t.SEnum.A
assert t.SEnum(1) is t.SEnum.B
assert t.SEnum(-1) is t.SEnum.C
assert t.from_enum(t.SEnum.A) == 0
assert t.from_enum(t.SEnum.B) == 1
assert t.from_enum(t.SEnum.C) == -1
def test03_enum_arithmetic():
assert t.SEnum.B + 2 == 3
assert t.SEnum.B + 2.5 == 3.5
assert 2 + t.SEnum.B == 3
assert 2.5 + t.SEnum.B == 3.5
assert t.SEnum.B >> t.SEnum.B == 0
assert t.SEnum.B << t.SEnum.B == 2
assert -t.SEnum.B == -1 and -t.SEnum.C == 1
assert t.SEnum.B & t.SEnum.B == 1
assert t.SEnum.B & ~t.SEnum.B == 0
with pytest.raises(TypeError, match="unsupported operand type"):
t.Enum.B + 2
with pytest.raises(TypeError, match="unsupported operand type"):
t.SEnum.B - "1"
with pytest.raises(TypeError, match="unsupported operand type"):
t.SEnum.B >> 1.0
def test04_enum_export():
assert t.Item1 is t.ClassicEnum.Item1 and t.Item1.value == 0
assert t.Item2 is t.ClassicEnum.Item2 and t.Item2.value == 1
# test for issue #39
def test05_enum_property():
w = t.EnumProperty()
assert w.read_enum == t.Enum.A
assert str(w.read_enum) == 'Enum.A'
def test08_enum_comparisons():
assert int(t.SEnum.B) == 1
for enum in (t.SEnum,):
value = getattr(enum, "B")
assert value != str(int(value))
assert value != int(value) + 0.4
assert value < int(value) + 0.4
for i in (0, 0.5, 1, 1.5, 2):
assert (value == i) == (int(value) == i)
assert (value != i) == (int(value) != i)
assert (value < i) == (int(value) < i)
assert (value <= i) == (int(value) <= i)
assert (value >= i) == (int(value) >= i)
assert (value > i) == (int(value) > i)
assert (i == value) == (i == int(value))
assert (i != value) == (i != int(value))
assert (i < value) == (i < int(value))
assert (i <= value) == (i <= int(value))
assert (i >= value) == (i >= int(value))
assert (i > value) == (i > int(value))
for unrelated in (None, "hello", "1"):
assert value != unrelated and unrelated != value
assert not (value == unrelated) and not (unrelated == value)
with pytest.raises(TypeError):
value < unrelated
with pytest.raises(TypeError):
unrelated < value
# different enum types never compare equal ...
assert t.Enum.B != t.SEnum.B and t.SEnum.B != t.Enum.B
assert not (t.Enum.B == t.SEnum.B) and not (t.SEnum.B == t.Enum.B)
assert t.Enum.B != t.SEnum.C and t.SEnum.C != t.Enum.B
def test06_enum_flag():
# repr / str tests
assert repr(t.Flag.A) == 'Flag.A'
assert str(t.Flag.A) == 'Flag.A'
assert repr(t.Flag.B) == 'Flag.B'
assert str(t.Flag.B) == 'Flag.B'
assert repr(t.Flag.C) == 'Flag.C'
assert str(t.Flag.C) == 'Flag.C'
assert repr(t.Flag.A | t.Flag.B) in ['Flag.A|B', 'Flag.B|A']
assert str(t.Flag.A | t.Flag.B) in ['Flag.A|B', 'Flag.B|A']
assert repr(t.Flag.A | t.Flag.B | t.Flag.C) in ['Flag.A|B|C', 'Flag.C|B|A']
assert str(t.Flag.A | t.Flag.B | t.Flag.C) in ['Flag.A|B|C', 'Flag.C|B|A']
# Flag membership tests
assert (t.Flag(1) | t.Flag(2)).value == 3
assert (t.Flag(3) & t.Flag(1)).value == 1
assert (t.Flag(3) ^ t.Flag(1)).value == 2
assert (t.Flag(3) == (t.Flag.A | t.Flag.B))
# ensure the flag mask is set correctly by enum_append in Python 3.11+
if hasattr(t.Flag, "_flag_mask_"):
assert t.Flag._flag_mask_ == 7
assert (t.from_enum(t.Flag.A | t.Flag.C) == 5)
assert (t.from_enum_implicit(t.Flag(1) | t.Flag(4)) == 5)
# unsigned flag tests to verify correct type casting behavior
# (in particular, overflow protection in enum_from_python.)
assert (t.UnsignedFlag(1) | t.UnsignedFlag(2)).value == 3
assert t.UnsignedFlag.A.value == 1
assert t.UnsignedFlag.B.value == 2
assert t.UnsignedFlag.All.value == 0xffffffffffffffff
assert t.UnsignedFlag(t.UnsignedFlag.A) is t.UnsignedFlag.A
assert t.UnsignedFlag(t.UnsignedFlag.B) is t.UnsignedFlag.B
assert t.UnsignedFlag(t.UnsignedFlag.All) is t.UnsignedFlag.All
assert t.from_enum(t.UnsignedFlag.A) == 1
assert t.from_enum(t.UnsignedFlag.B) == 2
assert t.from_enum(t.UnsignedFlag.All) == 0xffffffffffffffff
assert t.to_flag(1) == t.Flag.A
assert t.to_flag(2) == t.Flag.B
assert t.to_flag(4) == t.Flag.C
assert t.to_flag(5) == (t.Flag.A | t.Flag.C)
def test09_enum_methods():
assert t.Item1.my_value == 0 and t.Item2.my_value == 1
assert t.Item1.get_value() == 0 and t.Item2.get_value() == 1
assert t.Item1.foo() == t.Item1
assert t.ClassicEnum.bar(t.Item1) == t.Item1
def test10_enum_opaque():
assert t.OpaqueEnum.X == t.OpaqueEnum("X") and t.OpaqueEnum.Y == t.OpaqueEnum("Y")
|
