class MyIter(bt2._UserMessageIterator):
pass
- with self.assertRaises(bt2.IncompleteUserClass):
+ with self.assertRaises(bt2._IncompleteUserClass):
class MySource(bt2._UserSourceComponent):
pass
class MyIter(bt2._UserMessageIterator):
pass
- with self.assertRaises(bt2.IncompleteUserClass):
+ with self.assertRaises(bt2._IncompleteUserClass):
class MySource(bt2._UserSourceComponent, message_iterator_class=int):
pass
class MyIter(bt2._UserMessageIterator):
pass
- with self.assertRaises(bt2.IncompleteUserClass):
+ with self.assertRaises(bt2._IncompleteUserClass):
class MyFilter(bt2._UserFilterComponent):
pass
class MyIter(bt2._UserMessageIterator):
pass
- with self.assertRaises(bt2.IncompleteUserClass):
+ with self.assertRaises(bt2._IncompleteUserClass):
class MySink(bt2._UserSinkComponent):
pass
graph = bt2.Graph()
comp = graph.add_component(MySink, 'salut')
self._comp_cls = comp.cls
- self.assertTrue(issubclass(type(self._comp_cls), bt2.component._ComponentClass))
+ self.assertTrue(issubclass(type(self._comp_cls), bt2._SinkComponentClass))
def tearDown(self):
del self._py_comp_cls