+ def test_user_src_output_ports_getitem(self):
+ class MySource(
+ bt2._UserSourceComponent, message_iterator_class=bt2._UserMessageIterator
+ ):
+ def __init__(comp_self, config, params, obj):
+ port1 = comp_self._add_output_port("clear")
+ port2 = comp_self._add_output_port("print")
+ port3 = comp_self._add_output_port("insert")
+ self.assertEqual(port3.addr, comp_self._output_ports["insert"].addr)
+ self.assertEqual(port2.addr, comp_self._output_ports["print"].addr)
+ self.assertEqual(port1.addr, comp_self._output_ports["clear"].addr)
+
+ self._create_comp(MySource)
+
+ def test_user_flt_output_ports_getitem(self):
+ class MyFilter(
+ bt2._UserFilterComponent, message_iterator_class=bt2._UserMessageIterator
+ ):
+ def __init__(comp_self, config, params, obj):
+ port1 = comp_self._add_output_port("clear")
+ port2 = comp_self._add_output_port("print")
+ port3 = comp_self._add_output_port("insert")
+ self.assertEqual(port3.addr, comp_self._output_ports["insert"].addr)
+ self.assertEqual(port2.addr, comp_self._output_ports["print"].addr)
+ self.assertEqual(port1.addr, comp_self._output_ports["clear"].addr)
+
+ self._create_comp(MyFilter)
+
+ def test_user_flt_input_ports_getitem(self):
+ class MyFilter(
+ bt2._UserFilterComponent, message_iterator_class=bt2._UserMessageIterator
+ ):
+ def __init__(comp_self, config, params, obj):
+ port1 = comp_self._add_input_port("clear")
+ port2 = comp_self._add_input_port("print")
+ port3 = comp_self._add_input_port("insert")
+ self.assertEqual(port3.addr, comp_self._input_ports["insert"].addr)
+ self.assertEqual(port2.addr, comp_self._input_ports["print"].addr)
+ self.assertEqual(port1.addr, comp_self._input_ports["clear"].addr)
+
+ self._create_comp(MyFilter)