Commit | Line | Data |
---|---|---|
ae0bfae8 | 1 | from bt2 import value |
f6a5e476 PP |
2 | import collections |
3 | import unittest | |
4 | import copy | |
5 | import bt2 | |
6 | ||
7 | ||
871a292a | 8 | class _MyIter(bt2._UserMessageIterator): |
a4dcfa96 | 9 | def __init__(self, self_output_port): |
871a292a SM |
10 | self._build_meta() |
11 | self._at = 0 | |
12 | ||
13 | def _build_meta(self): | |
14 | self._tc = self._component._create_trace_class() | |
15 | self._t = self._tc() | |
16 | self._sc = self._tc.create_stream_class() | |
17 | self._ec = self._sc.create_event_class(name='salut') | |
18 | self._my_int_ft = self._tc.create_signed_integer_field_class(32) | |
19 | payload_ft = self._tc.create_structure_field_class() | |
20 | payload_ft += collections.OrderedDict([ | |
21 | ('my_int', self._my_int_ft), | |
22 | ]) | |
23 | self._ec.payload_field_type = payload_ft | |
24 | self._stream = self._t.create_stream(self._sc) | |
25 | self._packet = self._stream.create_packet() | |
26 | ||
27 | def _create_event(self, value): | |
28 | ev = self._ec() | |
29 | ev.payload_field['my_int'] = value | |
30 | ev.packet = self._packet | |
31 | return ev | |
32 | ||
33 | ||
f6a5e476 PP |
34 | class GraphTestCase(unittest.TestCase): |
35 | def setUp(self): | |
36 | self._graph = bt2.Graph() | |
37 | ||
38 | def tearDown(self): | |
39 | del self._graph | |
40 | ||
41 | def test_create_empty(self): | |
42 | graph = bt2.Graph() | |
43 | ||
44 | def test_add_component_user_cls(self): | |
45 | class MySink(bt2._UserSinkComponent): | |
46 | def _consume(self): | |
47 | pass | |
48 | ||
49 | comp = self._graph.add_component(MySink, 'salut') | |
50 | self.assertEqual(comp.name, 'salut') | |
51 | ||
52 | def test_add_component_gen_cls(self): | |
53 | class MySink(bt2._UserSinkComponent): | |
54 | def _consume(self): | |
55 | pass | |
56 | ||
57 | comp = self._graph.add_component(MySink, 'salut') | |
871a292a | 58 | assert comp |
c88be1c8 | 59 | comp2 = self._graph.add_component(comp.cls, 'salut2') |
f6a5e476 PP |
60 | self.assertEqual(comp2.name, 'salut2') |
61 | ||
62 | def test_add_component_params(self): | |
63 | comp_params = None | |
64 | ||
65 | class MySink(bt2._UserSinkComponent): | |
66 | def __init__(self, params): | |
67 | nonlocal comp_params | |
68 | comp_params = params | |
69 | ||
70 | def _consume(self): | |
71 | pass | |
72 | ||
73 | params = {'hello': 23, 'path': '/path/to/stuff'} | |
74 | comp = self._graph.add_component(MySink, 'salut', params) | |
75 | self.assertEqual(params, comp_params) | |
76 | del comp_params | |
77 | ||
78 | def test_add_component_invalid_cls_type(self): | |
79 | with self.assertRaises(TypeError): | |
80 | self._graph.add_component(int, 'salut') | |
81 | ||
78c432bb PP |
82 | def test_add_component_invalid_logging_level_type(self): |
83 | class MySink(bt2._UserSinkComponent): | |
84 | def _consume(self): | |
85 | pass | |
86 | ||
87 | with self.assertRaises(TypeError): | |
88 | self._graph.add_component(MySink, 'salut', logging_level='yo') | |
89 | ||
90 | def test_add_component_invalid_logging_level_value(self): | |
91 | class MySink(bt2._UserSinkComponent): | |
92 | def _consume(self): | |
93 | pass | |
94 | ||
95 | with self.assertRaises(ValueError): | |
96 | self._graph.add_component(MySink, 'salut', logging_level=12345) | |
97 | ||
98 | def test_add_component_logging_level(self): | |
99 | class MySink(bt2._UserSinkComponent): | |
100 | def _consume(self): | |
101 | pass | |
102 | ||
103 | comp = self._graph.add_component(MySink, 'salut', | |
104 | logging_level=bt2.LoggingLevel.DEBUG) | |
105 | self.assertEqual(comp.logging_level, bt2.LoggingLevel.DEBUG) | |
106 | ||
f6a5e476 | 107 | def test_connect_ports(self): |
fa4c33e3 | 108 | class MyIter(bt2._UserMessageIterator): |
f6a5e476 PP |
109 | def __next__(self): |
110 | raise bt2.Stop | |
111 | ||
112 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 113 | message_iterator_class=MyIter): |
f6a5e476 PP |
114 | def __init__(self, params): |
115 | self._add_output_port('out') | |
116 | ||
117 | class MySink(bt2._UserSinkComponent): | |
118 | def __init__(self, params): | |
119 | self._add_input_port('in') | |
120 | ||
121 | def _consume(self): | |
122 | raise bt2.Stop | |
123 | ||
124 | src = self._graph.add_component(MySource, 'src') | |
125 | sink = self._graph.add_component(MySink, 'sink') | |
871a292a | 126 | |
f6a5e476 PP |
127 | conn = self._graph.connect_ports(src.output_ports['out'], |
128 | sink.input_ports['in']) | |
129 | self.assertTrue(src.output_ports['out'].is_connected) | |
130 | self.assertTrue(sink.input_ports['in'].is_connected) | |
871a292a SM |
131 | self.assertEqual(src.output_ports['out'].connection._ptr, conn._ptr) |
132 | self.assertEqual(sink.input_ports['in'].connection._ptr, conn._ptr) | |
f6a5e476 PP |
133 | |
134 | def test_connect_ports_invalid_direction(self): | |
fa4c33e3 | 135 | class MyIter(bt2._UserMessageIterator): |
f6a5e476 PP |
136 | def __next__(self): |
137 | raise bt2.Stop | |
138 | ||
139 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 140 | message_iterator_class=MyIter): |
f6a5e476 PP |
141 | def __init__(self, params): |
142 | self._add_output_port('out') | |
143 | ||
144 | class MySink(bt2._UserSinkComponent): | |
145 | def __init__(self, params): | |
146 | self._add_input_port('in') | |
147 | ||
148 | def _consume(self): | |
149 | raise bt2.Stop | |
150 | ||
151 | src = self._graph.add_component(MySource, 'src') | |
152 | sink = self._graph.add_component(MySink, 'sink') | |
153 | ||
154 | with self.assertRaises(TypeError): | |
155 | conn = self._graph.connect_ports(sink.input_ports['in'], | |
156 | src.output_ports['out']) | |
157 | ||
158 | def test_connect_ports_refused(self): | |
fa4c33e3 | 159 | class MyIter(bt2._UserMessageIterator): |
f6a5e476 PP |
160 | def __next__(self): |
161 | raise bt2.Stop | |
162 | ||
163 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 164 | message_iterator_class=MyIter): |
f6a5e476 PP |
165 | def __init__(self, params): |
166 | self._add_output_port('out') | |
167 | ||
168 | class MySink(bt2._UserSinkComponent): | |
169 | def __init__(self, params): | |
170 | self._add_input_port('in') | |
171 | ||
172 | def _consume(self): | |
173 | raise bt2.Stop | |
174 | ||
175 | def _accept_port_connection(self, port, other_port): | |
176 | return False | |
177 | ||
178 | src = self._graph.add_component(MySource, 'src') | |
179 | sink = self._graph.add_component(MySink, 'sink') | |
180 | ||
181 | with self.assertRaises(bt2.PortConnectionRefused): | |
182 | conn = self._graph.connect_ports(src.output_ports['out'], | |
183 | sink.input_ports['in']) | |
184 | ||
871a292a SM |
185 | def test_cancel(self): |
186 | self.assertFalse(self._graph.is_canceled) | |
f6a5e476 | 187 | self._graph.cancel() |
871a292a | 188 | self.assertTrue(self._graph.is_canceled) |
f6a5e476 | 189 | |
871a292a SM |
190 | # Test that Graph.run() raises bt2.GraphCanceled if the graph gets canceled |
191 | # during execution. | |
192 | def test_cancel_while_running(self): | |
193 | class MyIter(_MyIter): | |
9ef22b36 | 194 | def __next__(self): |
871a292a | 195 | return self._create_stream_beginning_message(self._stream) |
9ef22b36 PP |
196 | |
197 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 198 | message_iterator_class=MyIter): |
9ef22b36 PP |
199 | def __init__(self, params): |
200 | self._add_output_port('out') | |
201 | ||
202 | class MySink(bt2._UserSinkComponent): | |
203 | def __init__(self, params): | |
204 | self._add_input_port('in') | |
205 | ||
206 | def _consume(self): | |
871a292a SM |
207 | # Pretend that somebody asynchronously cancelled the graph. |
208 | nonlocal graph | |
209 | graph.cancel() | |
9ef22b36 | 210 | |
871a292a | 211 | return next(self._msg_iter) |
9ef22b36 | 212 | |
871a292a SM |
213 | def _graph_is_configured(self): |
214 | self._msg_iter = self._input_ports['in'].create_message_iterator() | |
9ef22b36 | 215 | |
871a292a SM |
216 | graph = bt2.Graph() |
217 | up = graph.add_component(MySource, 'down') | |
218 | down = graph.add_component(MySink, 'up') | |
219 | graph.connect_ports(up.output_ports['out'], down.input_ports['in']) | |
220 | with self.assertRaises(bt2.GraphCanceled): | |
221 | graph.run() | |
f6a5e476 PP |
222 | |
223 | def test_run(self): | |
871a292a | 224 | class MyIter(_MyIter): |
f6a5e476 | 225 | def __next__(self): |
871a292a SM |
226 | if self._at == 9: |
227 | raise StopIteration | |
228 | ||
229 | if self._at == 0: | |
230 | msg = self._create_stream_beginning_message(self._stream) | |
231 | elif self._at == 1: | |
232 | msg = self._create_packet_beginning_message(self._packet) | |
233 | elif self._at == 7: | |
234 | msg = self._create_packet_end_message(self._packet) | |
235 | elif self._at == 8: | |
236 | msg = self._create_stream_end_message(self._stream) | |
237 | else: | |
238 | msg = self._create_event_message(self._ec, self._packet) | |
f6a5e476 | 239 | |
f6a5e476 | 240 | self._at += 1 |
fa4c33e3 | 241 | return msg |
f6a5e476 PP |
242 | |
243 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 244 | message_iterator_class=MyIter): |
f6a5e476 PP |
245 | def __init__(self, params): |
246 | self._add_output_port('out') | |
247 | ||
248 | class MySink(bt2._UserSinkComponent): | |
249 | def __init__(self, params): | |
871a292a | 250 | self._input_port = self._add_input_port('in') |
f6a5e476 PP |
251 | self._at = 0 |
252 | ||
253 | def _consume(comp_self): | |
fa4c33e3 | 254 | msg = next(comp_self._msg_iter) |
f6a5e476 PP |
255 | |
256 | if comp_self._at == 0: | |
871a292a | 257 | self.assertIsInstance(msg, bt2.message._StreamBeginningMessage) |
f6a5e476 | 258 | elif comp_self._at == 1: |
871a292a | 259 | self.assertIsInstance(msg, bt2.message._PacketBeginningMessage) |
f6a5e476 | 260 | elif comp_self._at >= 2 and comp_self._at <= 6: |
871a292a | 261 | self.assertIsInstance(msg, bt2.message._EventMessage) |
c88be1c8 | 262 | self.assertEqual(msg.event.cls.name, 'salut') |
f6a5e476 | 263 | elif comp_self._at == 7: |
871a292a | 264 | self.assertIsInstance(msg, bt2.message._PacketEndMessage) |
f6a5e476 | 265 | elif comp_self._at == 8: |
871a292a | 266 | self.assertIsInstance(msg, bt2.message._StreamEndMessage) |
f6a5e476 PP |
267 | |
268 | comp_self._at += 1 | |
269 | ||
871a292a SM |
270 | def _graph_is_configured(self): |
271 | self._msg_iter = self._input_port.create_message_iterator() | |
f6a5e476 PP |
272 | |
273 | src = self._graph.add_component(MySource, 'src') | |
274 | sink = self._graph.add_component(MySink, 'sink') | |
275 | conn = self._graph.connect_ports(src.output_ports['out'], | |
276 | sink.input_ports['in']) | |
277 | self._graph.run() | |
278 | ||
279 | def test_run_again(self): | |
871a292a | 280 | class MyIter(_MyIter): |
f6a5e476 | 281 | def __next__(self): |
871a292a | 282 | if self._at == 3: |
f6a5e476 PP |
283 | raise bt2.TryAgain |
284 | ||
871a292a SM |
285 | if self._at == 0: |
286 | msg = self._create_stream_beginning_message(self._stream) | |
287 | elif self._at == 1: | |
288 | msg = self._create_packet_beginning_message(self._packet) | |
289 | elif self._at == 2: | |
290 | msg = self._create_event_message(self._ec, self._packet) | |
291 | ||
f6a5e476 | 292 | self._at += 1 |
fa4c33e3 | 293 | return msg |
f6a5e476 PP |
294 | |
295 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 296 | message_iterator_class=MyIter): |
f6a5e476 PP |
297 | def __init__(self, params): |
298 | self._add_output_port('out') | |
299 | ||
300 | class MySink(bt2._UserSinkComponent): | |
301 | def __init__(self, params): | |
871a292a | 302 | self._input_port = self._add_input_port('in') |
f6a5e476 PP |
303 | self._at = 0 |
304 | ||
305 | def _consume(comp_self): | |
871a292a | 306 | msg = next(comp_self._msg_iter) |
f6a5e476 | 307 | if comp_self._at == 0: |
871a292a | 308 | self.assertIsInstance(msg, bt2.message._StreamBeginningMessage) |
f6a5e476 | 309 | elif comp_self._at == 1: |
871a292a SM |
310 | self.assertIsInstance(msg, bt2.message._PacketBeginningMessage) |
311 | elif comp_self._at == 2: | |
312 | self.assertIsInstance(msg, bt2.message._EventMessage) | |
f6a5e476 | 313 | raise bt2.TryAgain |
871a292a SM |
314 | else: |
315 | pass | |
f6a5e476 PP |
316 | |
317 | comp_self._at += 1 | |
318 | ||
871a292a SM |
319 | def _graph_is_configured(self): |
320 | self._msg_iter = self._input_port.create_message_iterator() | |
f6a5e476 PP |
321 | |
322 | src = self._graph.add_component(MySource, 'src') | |
323 | sink = self._graph.add_component(MySink, 'sink') | |
324 | conn = self._graph.connect_ports(src.output_ports['out'], | |
325 | sink.input_ports['in']) | |
326 | ||
327 | with self.assertRaises(bt2.TryAgain): | |
328 | self._graph.run() | |
329 | ||
f6a5e476 | 330 | def test_run_error(self): |
871a292a | 331 | raised_in_sink = False |
f6a5e476 | 332 | |
871a292a | 333 | class MyIter(_MyIter): |
f6a5e476 | 334 | def __next__(self): |
871a292a SM |
335 | # If this gets called after the sink raised an exception, it is |
336 | # an error. | |
337 | nonlocal raised_in_sink | |
338 | assert raised_in_sink is False | |
339 | ||
340 | if self._at == 0: | |
341 | msg = self._create_stream_beginning_message(self._stream) | |
342 | elif self._at == 1: | |
343 | msg = self._create_packet_beginning_message(self._packet) | |
344 | elif self._at == 2 or self._at == 3: | |
345 | msg = self._create_event_message(self._ec, self._packet) | |
346 | else: | |
f6a5e476 | 347 | raise bt2.TryAgain |
f6a5e476 | 348 | self._at += 1 |
fa4c33e3 | 349 | return msg |
f6a5e476 PP |
350 | |
351 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 352 | message_iterator_class=MyIter): |
f6a5e476 PP |
353 | def __init__(self, params): |
354 | self._add_output_port('out') | |
355 | ||
356 | class MySink(bt2._UserSinkComponent): | |
357 | def __init__(self, params): | |
871a292a | 358 | self._input_port = self._add_input_port('in') |
f6a5e476 PP |
359 | self._at = 0 |
360 | ||
361 | def _consume(comp_self): | |
871a292a | 362 | msg = next(comp_self._msg_iter) |
f6a5e476 | 363 | if comp_self._at == 0: |
871a292a | 364 | self.assertIsInstance(msg, bt2.message._StreamBeginningMessage) |
f6a5e476 | 365 | elif comp_self._at == 1: |
871a292a SM |
366 | self.assertIsInstance(msg, bt2.message._PacketBeginningMessage) |
367 | elif comp_self._at == 2: | |
368 | self.assertIsInstance(msg, bt2.message._EventMessage) | |
369 | elif comp_self._at == 3: | |
370 | nonlocal raised_in_sink | |
371 | raised_in_sink = True | |
f6a5e476 PP |
372 | raise RuntimeError('error!') |
373 | ||
374 | comp_self._at += 1 | |
375 | ||
871a292a SM |
376 | def _graph_is_configured(self): |
377 | self._msg_iter = self._input_port.create_message_iterator() | |
f6a5e476 PP |
378 | |
379 | src = self._graph.add_component(MySource, 'src') | |
380 | sink = self._graph.add_component(MySink, 'sink') | |
381 | conn = self._graph.connect_ports(src.output_ports['out'], | |
382 | sink.input_ports['in']) | |
383 | ||
384 | with self.assertRaises(bt2.Error): | |
385 | self._graph.run() | |
386 | ||
871a292a | 387 | def test_listeners(self): |
fa4c33e3 | 388 | class MyIter(bt2._UserMessageIterator): |
871a292a SM |
389 | def __next__(self): |
390 | raise bt2.Stop | |
9ef22b36 PP |
391 | |
392 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 393 | message_iterator_class=MyIter): |
9ef22b36 PP |
394 | def __init__(self, params): |
395 | self._add_output_port('out') | |
871a292a | 396 | self._add_output_port('zero') |
9ef22b36 PP |
397 | |
398 | class MySink(bt2._UserSinkComponent): | |
399 | def __init__(self, params): | |
400 | self._add_input_port('in') | |
9ef22b36 | 401 | |
871a292a SM |
402 | def _consume(self): |
403 | raise bt2.Stop | |
9ef22b36 | 404 | |
871a292a SM |
405 | def _port_connected(self, port, other_port): |
406 | self._add_input_port('taste') | |
9ef22b36 | 407 | |
871a292a SM |
408 | def port_added_listener(component, port): |
409 | nonlocal calls | |
410 | calls.append((port_added_listener, component, port)) | |
9ef22b36 | 411 | |
871a292a SM |
412 | def ports_connected_listener(upstream_component, upstream_port, |
413 | downstream_component, downstream_port): | |
414 | nonlocal calls | |
415 | calls.append((ports_connected_listener, | |
416 | upstream_component, upstream_port, | |
417 | downstream_component, downstream_port)) | |
418 | ||
419 | calls = [] | |
420 | self._graph.add_port_added_listener(port_added_listener) | |
421 | self._graph.add_ports_connected_listener(ports_connected_listener) | |
9ef22b36 PP |
422 | src = self._graph.add_component(MySource, 'src') |
423 | sink = self._graph.add_component(MySink, 'sink') | |
871a292a SM |
424 | self._graph.connect_ports(src.output_ports['out'], |
425 | sink.input_ports['in']) | |
9ef22b36 | 426 | |
871a292a SM |
427 | self.assertEqual(len(calls), 5) |
428 | ||
429 | self.assertIs(calls[0][0], port_added_listener) | |
430 | self.assertEqual(calls[0][1].name, 'src') | |
431 | self.assertEqual(calls[0][2].name, 'out') | |
432 | ||
433 | self.assertIs(calls[1][0], port_added_listener) | |
434 | self.assertEqual(calls[1][1].name, 'src') | |
435 | self.assertEqual(calls[1][2].name, 'zero') | |
436 | ||
437 | self.assertIs(calls[2][0], port_added_listener) | |
438 | self.assertEqual(calls[2][1].name, 'sink') | |
439 | self.assertEqual(calls[2][2].name, 'in') | |
440 | ||
441 | self.assertIs(calls[3][0], port_added_listener) | |
442 | self.assertEqual(calls[3][1].name, 'sink') | |
443 | self.assertEqual(calls[3][2].name, 'taste') | |
444 | ||
445 | self.assertIs(calls[4][0], ports_connected_listener) | |
446 | self.assertEqual(calls[4][1].name, 'src') | |
447 | self.assertEqual(calls[4][2].name, 'out') | |
448 | self.assertEqual(calls[4][3].name, 'sink') | |
449 | self.assertEqual(calls[4][4].name, 'in') | |
450 | ||
451 | def test_invalid_listeners(self): | |
fa4c33e3 | 452 | class MyIter(bt2._UserMessageIterator): |
f6a5e476 PP |
453 | def __next__(self): |
454 | raise bt2.Stop | |
455 | ||
456 | class MySource(bt2._UserSourceComponent, | |
fa4c33e3 | 457 | message_iterator_class=MyIter): |
f6a5e476 PP |
458 | def __init__(self, params): |
459 | self._add_output_port('out') | |
460 | self._add_output_port('zero') | |
461 | ||
f6a5e476 PP |
462 | class MySink(bt2._UserSinkComponent): |
463 | def __init__(self, params): | |
464 | self._add_input_port('in') | |
465 | ||
466 | def _consume(self): | |
467 | raise bt2.Stop | |
468 | ||
469 | def _port_connected(self, port, other_port): | |
470 | self._add_input_port('taste') | |
471 | ||
871a292a SM |
472 | with self.assertRaises(TypeError): |
473 | self._graph.add_port_added_listener(1234) | |
474 | with self.assertRaises(TypeError): | |
475 | self._graph.add_ports_connected_listener(1234) | |
f6a5e476 | 476 | |
871a292a SM |
477 | def test_raise_in_component_init(self): |
478 | class MySink(bt2._UserSinkComponent): | |
479 | def __init__(self, params): | |
480 | raise ValueError('oops!') | |
f6a5e476 | 481 | |
871a292a SM |
482 | def _consume(self): |
483 | raise bt2.Stop | |
484 | ||
485 | graph = bt2.Graph() | |
486 | ||
487 | with self.assertRaises(bt2.Error): | |
488 | graph.add_component(MySink, 'comp') | |
489 | ||
490 | def test_raise_in_port_added_listener(self): | |
491 | class MySink(bt2._UserSinkComponent): | |
492 | def __init__(self, params): | |
493 | self._add_input_port('in') | |
494 | ||
495 | def _consume(self): | |
496 | raise bt2.Stop | |
497 | ||
498 | def port_added_listener(component, port): | |
499 | raise ValueError('oh noes!') | |
500 | ||
501 | graph = bt2.Graph() | |
502 | graph.add_port_added_listener(port_added_listener) | |
503 | ||
504 | with self.assertRaises(bt2.Error): | |
505 | graph.add_component(MySink, 'comp') | |
506 | ||
507 | def test_raise_in_ports_connected_listener(self): | |
508 | class MyIter(bt2._UserMessageIterator): | |
509 | def __next__(self): | |
510 | raise bt2.Stop | |
511 | ||
512 | class MySource(bt2._UserSourceComponent, | |
513 | message_iterator_class=MyIter): | |
514 | def __init__(self, params): | |
515 | self._add_output_port('out') | |
516 | ||
517 | class MySink(bt2._UserSinkComponent): | |
518 | def __init__(self, params): | |
519 | self._add_input_port('in') | |
520 | ||
521 | def _consume(self): | |
522 | raise bt2.Stop | |
f6a5e476 | 523 | |
a4dcfa96 SM |
524 | def ports_connected_listener(upstream_component, upstream_port, |
525 | downstream_component, downstream_port): | |
871a292a | 526 | raise ValueError('oh noes!') |
f6a5e476 | 527 | |
871a292a SM |
528 | graph = bt2.Graph() |
529 | graph.add_ports_connected_listener(ports_connected_listener) | |
530 | up = graph.add_component(MySource, 'down') | |
531 | down = graph.add_component(MySink, 'up') | |
f6a5e476 | 532 | |
871a292a SM |
533 | with self.assertRaises(bt2.Error): |
534 | graph.connect_ports(up.output_ports['out'], down.input_ports['in']) |