Commit | Line | Data |
---|---|---|
32d2d479 MJ |
1 | # |
2 | # Copyright (C) 2019 EfficiOS Inc. | |
3 | # | |
4 | # This program is free software; you can redistribute it and/or | |
5 | # modify it under the terms of the GNU General Public License | |
6 | # as published by the Free Software Foundation; only version 2 | |
7 | # of the License. | |
8 | # | |
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU General Public License for more details. | |
13 | # | |
14 | # You should have received a copy of the GNU General Public License | |
15 | # along with this program; if not, write to the Free Software | |
16 | # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | |
17 | # | |
18 | ||
9cf643d1 | 19 | import unittest |
060aee52 | 20 | from utils import run_in_component_init |
9cbe0c59 FD |
21 | from bt2 import stream_class as bt2_stream_class |
22 | from bt2 import trace_class as bt2_trace_class | |
23 | from bt2 import clock_class as bt2_clock_class | |
24 | from bt2 import event_class as bt2_event_class | |
25 | from bt2 import field_class as bt2_field_class | |
9cf643d1 PP |
26 | |
27 | ||
28 | class StreamClassTestCase(unittest.TestCase): | |
29 | def setUp(self): | |
060aee52 SM |
30 | def f(comp_self): |
31 | tc = comp_self._create_trace_class(assigns_automatic_stream_class_id=True) | |
32 | cc = comp_self._create_clock_class() | |
33 | return tc, cc | |
34 | ||
35 | self._tc, self._cc = run_in_component_init(f) | |
36 | self._trace = self._tc() | |
37 | ||
38 | def test_create_default(self): | |
39 | sc = self._tc.create_stream_class() | |
40 | ||
9cbe0c59 | 41 | self.assertIs(type(sc), bt2_stream_class._StreamClass) |
060aee52 SM |
42 | self.assertIsNone(sc.name) |
43 | self.assertIsNone(sc.packet_context_field_class) | |
44 | self.assertIsNone(sc.event_common_context_field_class) | |
45 | self.assertIsNone(sc.default_clock_class) | |
46 | self.assertTrue(sc.assigns_automatic_event_class_id) | |
47 | self.assertTrue(sc.assigns_automatic_stream_id) | |
37a93d41 | 48 | self.assertFalse(sc.supports_packets) |
5ef34326 PP |
49 | self.assertFalse(sc.packets_have_beginning_default_clock_snapshot) |
50 | self.assertFalse(sc.packets_have_end_default_clock_snapshot) | |
77037b2b PP |
51 | self.assertFalse(sc.supports_discarded_events) |
52 | self.assertFalse(sc.discarded_events_have_default_clock_snapshots) | |
53 | self.assertFalse(sc.supports_discarded_packets) | |
54 | self.assertFalse(sc.discarded_packets_have_default_clock_snapshots) | |
b2df5780 | 55 | self.assertEqual(len(sc.user_attributes), 0) |
060aee52 SM |
56 | |
57 | def test_create_name(self): | |
58 | sc = self._tc.create_stream_class(name='bozo') | |
59 | self.assertEqual(sc.name, 'bozo') | |
60 | ||
61 | def test_create_invalid_name(self): | |
d5a22ce8 | 62 | with self.assertRaisesRegex(TypeError, "'int' is not a 'str' object"): |
060aee52 | 63 | self._tc.create_stream_class(name=17) |
9cf643d1 | 64 | |
a373e036 SM |
65 | self.assertEqual(len(self._tc), 0) |
66 | ||
060aee52 SM |
67 | def test_create_packet_context_field_class(self): |
68 | fc = self._tc.create_structure_field_class() | |
61d96b89 FD |
69 | sc = self._tc.create_stream_class( |
70 | packet_context_field_class=fc, supports_packets=True | |
71 | ) | |
060aee52 | 72 | self.assertEqual(sc.packet_context_field_class, fc) |
9cbe0c59 FD |
73 | self.assertIs( |
74 | type(sc.packet_context_field_class), bt2_field_class._StructureFieldClass | |
75 | ) | |
9cf643d1 | 76 | |
060aee52 | 77 | def test_create_invalid_packet_context_field_class(self): |
d5a22ce8 SM |
78 | with self.assertRaisesRegex( |
79 | TypeError, | |
80 | "'int' is not a '<class 'bt2.field_class._StructureFieldClass'>' object", | |
81 | ): | |
a373e036 SM |
82 | self._tc.create_stream_class( |
83 | packet_context_field_class=22, supports_packets=True | |
84 | ) | |
85 | ||
86 | self.assertEqual(len(self._tc), 0) | |
9cf643d1 | 87 | |
37a93d41 PP |
88 | def test_create_invalid_packet_context_field_class_no_packets(self): |
89 | fc = self._tc.create_structure_field_class() | |
90 | ||
d5a22ce8 SM |
91 | with self.assertRaisesRegex( |
92 | ValueError, | |
93 | "cannot have a packet context field class without supporting packets", | |
94 | ): | |
37a93d41 PP |
95 | self._tc.create_stream_class(packet_context_field_class=fc) |
96 | ||
a373e036 SM |
97 | self.assertEqual(len(self._tc), 0) |
98 | ||
060aee52 SM |
99 | def test_create_event_common_context_field_class(self): |
100 | fc = self._tc.create_structure_field_class() | |
101 | sc = self._tc.create_stream_class(event_common_context_field_class=fc) | |
102 | self.assertEqual(sc.event_common_context_field_class, fc) | |
9cbe0c59 FD |
103 | self.assertIs( |
104 | type(sc.event_common_context_field_class), | |
105 | bt2_field_class._StructureFieldClass, | |
106 | ) | |
f6a5e476 | 107 | |
060aee52 | 108 | def test_create_invalid_event_common_context_field_class(self): |
d5a22ce8 SM |
109 | with self.assertRaisesRegex( |
110 | TypeError, | |
111 | "'int' is not a '<class 'bt2.field_class._StructureFieldClass'>' object", | |
112 | ): | |
060aee52 | 113 | self._tc.create_stream_class(event_common_context_field_class=22) |
9cf643d1 | 114 | |
a373e036 SM |
115 | self.assertEqual(len(self._tc), 0) |
116 | ||
060aee52 SM |
117 | def test_create_default_clock_class(self): |
118 | sc = self._tc.create_stream_class(default_clock_class=self._cc) | |
119 | self.assertEqual(sc.default_clock_class.addr, self._cc.addr) | |
9cbe0c59 | 120 | self.assertIs(type(sc.default_clock_class), bt2_clock_class._ClockClass) |
9cf643d1 | 121 | |
060aee52 | 122 | def test_create_invalid_default_clock_class(self): |
d5a22ce8 SM |
123 | with self.assertRaisesRegex( |
124 | TypeError, "'int' is not a '<class 'bt2.clock_class._ClockClass'>' object" | |
125 | ): | |
060aee52 | 126 | self._tc.create_stream_class(default_clock_class=12) |
9cf643d1 | 127 | |
a373e036 SM |
128 | self.assertEqual(len(self._tc), 0) |
129 | ||
b2df5780 PP |
130 | def test_create_user_attributes(self): |
131 | sc = self._tc.create_stream_class(user_attributes={'salut': 23}) | |
132 | self.assertEqual(sc.user_attributes, {'salut': 23}) | |
133 | ||
134 | def test_create_invalid_user_attributes(self): | |
d5a22ce8 SM |
135 | with self.assertRaisesRegex( |
136 | TypeError, "cannot create value object from 'object' object" | |
137 | ): | |
b2df5780 PP |
138 | self._tc.create_stream_class(user_attributes=object()) |
139 | ||
a373e036 SM |
140 | self.assertEqual(len(self._tc), 0) |
141 | ||
b2df5780 | 142 | def test_create_invalid_user_attributes_value_type(self): |
d5a22ce8 SM |
143 | with self.assertRaisesRegex( |
144 | TypeError, | |
145 | "'SignedIntegerValue' is not a '<class 'bt2.value.MapValue'>' object", | |
146 | ): | |
b2df5780 PP |
147 | self._tc.create_stream_class(user_attributes=23) |
148 | ||
a373e036 SM |
149 | self.assertEqual(len(self._tc), 0) |
150 | ||
060aee52 SM |
151 | def test_automatic_stream_ids(self): |
152 | sc = self._tc.create_stream_class(assigns_automatic_stream_id=True) | |
153 | self.assertTrue(sc.assigns_automatic_stream_id) | |
9cf643d1 | 154 | |
060aee52 SM |
155 | stream = self._trace.create_stream(sc) |
156 | self.assertIsNotNone(stream.id) | |
9cf643d1 | 157 | |
060aee52 SM |
158 | def test_automatic_stream_ids_raises(self): |
159 | sc = self._tc.create_stream_class(assigns_automatic_stream_id=True) | |
160 | self.assertTrue(sc.assigns_automatic_stream_id) | |
161 | ||
d5a22ce8 SM |
162 | with self.assertRaisesRegex( |
163 | ValueError, "id provided, but stream class assigns automatic stream ids" | |
164 | ): | |
060aee52 SM |
165 | self._trace.create_stream(sc, id=123) |
166 | ||
a373e036 SM |
167 | self.assertEqual(len(self._trace), 0) |
168 | ||
169 | def test_automatic_stream_ids_wrong_type(self): | |
170 | with self.assertRaisesRegex(TypeError, "str' is not a 'bool' object"): | |
171 | self._tc.create_stream_class(assigns_automatic_stream_id='True') | |
172 | ||
173 | self.assertEqual(len(self._tc), 0) | |
174 | ||
060aee52 SM |
175 | def test_no_automatic_stream_ids(self): |
176 | sc = self._tc.create_stream_class(assigns_automatic_stream_id=False) | |
177 | self.assertFalse(sc.assigns_automatic_stream_id) | |
178 | ||
179 | stream = self._trace.create_stream(sc, id=333) | |
180 | self.assertEqual(stream.id, 333) | |
181 | ||
182 | def test_no_automatic_stream_ids_raises(self): | |
183 | sc = self._tc.create_stream_class(assigns_automatic_stream_id=False) | |
184 | self.assertFalse(sc.assigns_automatic_stream_id) | |
185 | ||
d5a22ce8 SM |
186 | with self.assertRaisesRegex( |
187 | ValueError, | |
188 | "id not provided, but stream class does not assign automatic stream ids", | |
189 | ): | |
060aee52 SM |
190 | self._trace.create_stream(sc) |
191 | ||
a373e036 SM |
192 | self.assertEqual(len(self._trace), 0) |
193 | ||
060aee52 SM |
194 | def test_automatic_event_class_ids(self): |
195 | sc = self._tc.create_stream_class(assigns_automatic_event_class_id=True) | |
196 | self.assertTrue(sc.assigns_automatic_event_class_id) | |
197 | ||
198 | ec = sc.create_event_class() | |
199 | self.assertIsNotNone(ec.id) | |
200 | ||
201 | def test_automatic_event_class_ids_raises(self): | |
202 | sc = self._tc.create_stream_class(assigns_automatic_event_class_id=True) | |
203 | self.assertTrue(sc.assigns_automatic_event_class_id) | |
204 | ||
d5a22ce8 SM |
205 | with self.assertRaisesRegex( |
206 | ValueError, | |
207 | "id provided, but stream class assigns automatic event class ids", | |
208 | ): | |
060aee52 SM |
209 | sc.create_event_class(id=123) |
210 | ||
a373e036 SM |
211 | self.assertEqual(len(sc), 0) |
212 | ||
213 | def test_automatic_event_class_ids_wrong_type(self): | |
214 | with self.assertRaisesRegex(TypeError, "'str' is not a 'bool' object"): | |
215 | self._tc.create_stream_class(assigns_automatic_event_class_id='True') | |
216 | ||
217 | self.assertEqual(len(self._tc), 0) | |
218 | ||
060aee52 SM |
219 | def test_no_automatic_event_class_ids(self): |
220 | sc = self._tc.create_stream_class(assigns_automatic_event_class_id=False) | |
221 | self.assertFalse(sc.assigns_automatic_event_class_id) | |
9cf643d1 | 222 | |
060aee52 SM |
223 | ec = sc.create_event_class(id=333) |
224 | self.assertEqual(ec.id, 333) | |
9cf643d1 | 225 | |
060aee52 SM |
226 | def test_no_automatic_event_class_ids_raises(self): |
227 | sc = self._tc.create_stream_class(assigns_automatic_event_class_id=False) | |
228 | self.assertFalse(sc.assigns_automatic_event_class_id) | |
9cf643d1 | 229 | |
d5a22ce8 SM |
230 | with self.assertRaisesRegex( |
231 | ValueError, | |
232 | "id not provided, but stream class does not assign automatic event class ids", | |
233 | ): | |
060aee52 SM |
234 | sc.create_event_class() |
235 | ||
a373e036 SM |
236 | self.assertEqual(len(sc), 0) |
237 | ||
37a93d41 | 238 | def test_supports_packets_without_cs(self): |
61d96b89 FD |
239 | sc = self._tc.create_stream_class( |
240 | default_clock_class=self._cc, supports_packets=True | |
241 | ) | |
37a93d41 PP |
242 | self.assertTrue(sc.supports_packets) |
243 | self.assertFalse(sc.packets_have_beginning_default_clock_snapshot) | |
244 | self.assertFalse(sc.packets_have_end_default_clock_snapshot) | |
245 | ||
246 | def test_supports_packets_with_begin_cs(self): | |
61d96b89 FD |
247 | sc = self._tc.create_stream_class( |
248 | default_clock_class=self._cc, | |
249 | supports_packets=True, | |
250 | packets_have_beginning_default_clock_snapshot=True, | |
251 | ) | |
37a93d41 | 252 | self.assertTrue(sc.supports_packets) |
5ef34326 | 253 | self.assertTrue(sc.packets_have_beginning_default_clock_snapshot) |
37a93d41 | 254 | self.assertFalse(sc.packets_have_end_default_clock_snapshot) |
060aee52 | 255 | |
37a93d41 | 256 | def test_supports_packets_with_end_cs(self): |
61d96b89 FD |
257 | sc = self._tc.create_stream_class( |
258 | default_clock_class=self._cc, | |
259 | supports_packets=True, | |
260 | packets_have_end_default_clock_snapshot=True, | |
261 | ) | |
37a93d41 PP |
262 | self.assertTrue(sc.supports_packets) |
263 | self.assertFalse(sc.packets_have_beginning_default_clock_snapshot) | |
264 | self.assertTrue(sc.packets_have_end_default_clock_snapshot) | |
265 | ||
266 | def test_supports_packets_raises_type_error(self): | |
d5a22ce8 | 267 | with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): |
36153ada | 268 | self._tc.create_stream_class( |
61d96b89 FD |
269 | default_clock_class=self._cc, supports_packets=23 |
270 | ) | |
060aee52 | 271 | |
a373e036 SM |
272 | self.assertEqual(len(self._tc), 0) |
273 | ||
37a93d41 | 274 | def test_packets_have_begin_default_cs_raises_type_error(self): |
d5a22ce8 | 275 | with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): |
36153ada | 276 | self._tc.create_stream_class( |
61d96b89 FD |
277 | default_clock_class=self._cc, |
278 | packets_have_beginning_default_clock_snapshot=23, | |
279 | ) | |
060aee52 | 280 | |
a373e036 SM |
281 | self.assertEqual(len(self._tc), 0) |
282 | ||
37a93d41 | 283 | def test_packets_have_end_default_cs_raises_type_error(self): |
d5a22ce8 | 284 | with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): |
36153ada | 285 | self._tc.create_stream_class( |
61d96b89 FD |
286 | default_clock_class=self._cc, packets_have_end_default_clock_snapshot=23 |
287 | ) | |
37a93d41 | 288 | |
a373e036 SM |
289 | self.assertEqual(len(self._tc), 0) |
290 | ||
37a93d41 | 291 | def test_does_not_support_packets_raises_with_begin_cs(self): |
d5a22ce8 SM |
292 | with self.assertRaisesRegex( |
293 | ValueError, | |
294 | "cannot not support packets, but have packet beginning default clock snapshot", | |
295 | ): | |
36153ada | 296 | self._tc.create_stream_class( |
61d96b89 FD |
297 | default_clock_class=self._cc, |
298 | packets_have_beginning_default_clock_snapshot=True, | |
299 | ) | |
37a93d41 | 300 | |
a373e036 SM |
301 | self.assertEqual(len(self._tc), 0) |
302 | ||
37a93d41 | 303 | def test_does_not_support_packets_raises_with_end_cs(self): |
d5a22ce8 SM |
304 | with self.assertRaisesRegex( |
305 | ValueError, | |
306 | "cannot not support packets, but have packet end default clock snapshots", | |
307 | ): | |
36153ada | 308 | self._tc.create_stream_class( |
61d96b89 FD |
309 | default_clock_class=self._cc, |
310 | packets_have_end_default_clock_snapshot=True, | |
311 | ) | |
060aee52 | 312 | |
a373e036 SM |
313 | self.assertEqual(len(self._tc), 0) |
314 | ||
77037b2b | 315 | def test_supports_discarded_events_without_cs(self): |
61d96b89 FD |
316 | sc = self._tc.create_stream_class( |
317 | default_clock_class=self._cc, supports_discarded_events=True | |
318 | ) | |
77037b2b PP |
319 | self.assertTrue(sc.supports_discarded_events) |
320 | self.assertFalse(sc.discarded_events_have_default_clock_snapshots) | |
321 | ||
322 | def test_supports_discarded_events_with_cs(self): | |
61d96b89 FD |
323 | sc = self._tc.create_stream_class( |
324 | default_clock_class=self._cc, | |
325 | supports_discarded_events=True, | |
326 | discarded_events_have_default_clock_snapshots=True, | |
327 | ) | |
77037b2b PP |
328 | self.assertTrue(sc.supports_discarded_events) |
329 | self.assertTrue(sc.discarded_events_have_default_clock_snapshots) | |
330 | ||
331 | def test_supports_discarded_events_raises_type_error(self): | |
d5a22ce8 | 332 | with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): |
36153ada | 333 | self._tc.create_stream_class( |
61d96b89 FD |
334 | default_clock_class=self._cc, supports_discarded_events=23 |
335 | ) | |
77037b2b | 336 | |
a373e036 SM |
337 | self.assertEqual(len(self._tc), 0) |
338 | ||
77037b2b | 339 | def test_discarded_events_have_default_cs_raises_type_error(self): |
d5a22ce8 | 340 | with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): |
36153ada | 341 | self._tc.create_stream_class( |
61d96b89 FD |
342 | default_clock_class=self._cc, |
343 | discarded_events_have_default_clock_snapshots=23, | |
344 | ) | |
77037b2b | 345 | |
a373e036 SM |
346 | self.assertEqual(len(self._tc), 0) |
347 | ||
77037b2b | 348 | def test_does_not_support_discarded_events_raises_with_cs(self): |
d5a22ce8 SM |
349 | with self.assertRaisesRegex( |
350 | ValueError, | |
351 | "cannot not support discarded events, but have default clock snapshots for discarded event messages", | |
352 | ): | |
36153ada | 353 | self._tc.create_stream_class( |
61d96b89 FD |
354 | default_clock_class=self._cc, |
355 | discarded_events_have_default_clock_snapshots=True, | |
356 | ) | |
77037b2b | 357 | |
a373e036 SM |
358 | self.assertEqual(len(self._tc), 0) |
359 | ||
77037b2b | 360 | def test_supports_discarded_packets_without_cs(self): |
61d96b89 FD |
361 | sc = self._tc.create_stream_class( |
362 | default_clock_class=self._cc, | |
363 | supports_discarded_packets=True, | |
364 | supports_packets=True, | |
365 | ) | |
77037b2b PP |
366 | self.assertTrue(sc.supports_discarded_packets) |
367 | self.assertFalse(sc.discarded_packets_have_default_clock_snapshots) | |
368 | ||
369 | def test_supports_discarded_packets_with_cs(self): | |
61d96b89 FD |
370 | sc = self._tc.create_stream_class( |
371 | default_clock_class=self._cc, | |
372 | supports_discarded_packets=True, | |
373 | discarded_packets_have_default_clock_snapshots=True, | |
374 | supports_packets=True, | |
375 | ) | |
77037b2b PP |
376 | self.assertTrue(sc.supports_discarded_packets) |
377 | self.assertTrue(sc.discarded_packets_have_default_clock_snapshots) | |
378 | ||
37a93d41 | 379 | def test_supports_discarded_packets_raises_without_packet_support(self): |
d5a22ce8 SM |
380 | with self.assertRaisesRegex( |
381 | ValueError, "cannot support discarded packets, but not support packets" | |
382 | ): | |
36153ada | 383 | self._tc.create_stream_class( |
61d96b89 FD |
384 | default_clock_class=self._cc, supports_discarded_packets=True |
385 | ) | |
37a93d41 | 386 | |
a373e036 SM |
387 | self.assertEqual(len(self._tc), 0) |
388 | ||
77037b2b | 389 | def test_supports_discarded_packets_raises_type_error(self): |
d5a22ce8 | 390 | with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): |
36153ada | 391 | self._tc.create_stream_class( |
61d96b89 FD |
392 | default_clock_class=self._cc, |
393 | supports_discarded_packets=23, | |
394 | supports_packets=True, | |
395 | ) | |
77037b2b | 396 | |
a373e036 SM |
397 | self.assertEqual(len(self._tc), 0) |
398 | ||
77037b2b | 399 | def test_discarded_packets_have_default_cs_raises_type_error(self): |
d5a22ce8 | 400 | with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): |
36153ada | 401 | self._tc.create_stream_class( |
61d96b89 FD |
402 | default_clock_class=self._cc, |
403 | discarded_packets_have_default_clock_snapshots=23, | |
404 | supports_packets=True, | |
405 | ) | |
77037b2b | 406 | |
a373e036 SM |
407 | self.assertEqual(len(self._tc), 0) |
408 | ||
77037b2b | 409 | def test_does_not_support_discarded_packets_raises_with_cs(self): |
d5a22ce8 SM |
410 | with self.assertRaisesRegex( |
411 | ValueError, | |
412 | "cannot not support discarded packets, but have default clock snapshots for discarded packet messages", | |
413 | ): | |
36153ada | 414 | self._tc.create_stream_class( |
61d96b89 FD |
415 | default_clock_class=self._cc, |
416 | discarded_packets_have_default_clock_snapshots=True, | |
417 | supports_packets=True, | |
418 | ) | |
77037b2b | 419 | |
a373e036 SM |
420 | self.assertEqual(len(self._tc), 0) |
421 | ||
060aee52 SM |
422 | def test_trace_class(self): |
423 | sc = self._tc.create_stream_class() | |
424 | self.assertEqual(sc.trace_class.addr, self._tc.addr) | |
9cbe0c59 | 425 | self.assertIs(type(sc.trace_class), bt2_trace_class._TraceClass) |
060aee52 SM |
426 | |
427 | def _create_stream_class_with_event_classes(self): | |
428 | sc = self._tc.create_stream_class(assigns_automatic_event_class_id=False) | |
429 | ec1 = sc.create_event_class(id=23) | |
430 | ec2 = sc.create_event_class(id=17) | |
431 | return sc, ec1, ec2 | |
9cf643d1 PP |
432 | |
433 | def test_getitem(self): | |
060aee52 SM |
434 | sc, ec1, ec2 = self._create_stream_class_with_event_classes() |
435 | ||
436 | self.assertEqual(sc[23].addr, ec1.addr) | |
9cbe0c59 | 437 | self.assertEqual(type(sc[23]), bt2_event_class._EventClass) |
060aee52 | 438 | self.assertEqual(sc[17].addr, ec2.addr) |
9cbe0c59 | 439 | self.assertEqual(type(sc[17]), bt2_event_class._EventClass) |
9cf643d1 PP |
440 | |
441 | def test_getitem_wrong_key_type(self): | |
060aee52 SM |
442 | sc, _, _ = self._create_stream_class_with_event_classes() |
443 | ||
d5a22ce8 | 444 | with self.assertRaisesRegex(TypeError, "'str' is not an 'int' object"): |
060aee52 | 445 | sc['event23'] |
9cf643d1 PP |
446 | |
447 | def test_getitem_wrong_key(self): | |
060aee52 SM |
448 | sc, _, _ = self._create_stream_class_with_event_classes() |
449 | ||
d5a22ce8 | 450 | with self.assertRaisesRegex(KeyError, '19'): |
060aee52 | 451 | sc[19] |
9cf643d1 PP |
452 | |
453 | def test_len(self): | |
060aee52 SM |
454 | sc, _, _ = self._create_stream_class_with_event_classes() |
455 | ||
456 | self.assertEqual(len(sc), 2) | |
9cf643d1 PP |
457 | |
458 | def test_iter(self): | |
060aee52 SM |
459 | sc, _, _ = self._create_stream_class_with_event_classes() |
460 | ||
461 | ec_ids = sorted(sc) | |
462 | self.assertEqual(ec_ids, [17, 23]) | |
3db06b1d SM |
463 | |
464 | ||
465 | if __name__ == '__main__': | |
466 | unittest.main() |