7 class UserComponentTestCase(unittest
.TestCase
):
9 def _create_comp(comp_cls
, name
=None):
15 return graph
.add_component(comp_cls
, name
)
18 class MySink(bt2
._UserSinkComponent
):
19 def __init__(comp_self
, params
):
20 self
.assertEqual(comp_self
.name
, 'yaes')
25 comp
= self
._create
_comp
(MySink
, 'yaes')
28 class MySink(bt2
._UserSinkComponent
):
29 def __init__(comp_self
, params
):
31 self
.assertEqual(comp_self
.graph
, graph
)
37 comp
= graph
.add_component(MySink
, 'lel')
41 class MySink(bt2
._UserSinkComponent
):
42 def __init__(comp_self
, params
):
43 self
.assertEqual(comp_self
.component_class
, MySink
)
48 self
._create
_comp
(MySink
)
51 class MySink(bt2
._UserSinkComponent
):
52 def __init__(comp_self
, params
):
53 self
.assertIsInstance(comp_self
.addr
, int)
54 self
.assertNotEqual(comp_self
.addr
, 0)
59 self
._create
_comp
(MySink
)
61 def test_finalize(self
):
64 class MySink(bt2
._UserSinkComponent
):
68 def _finalize(comp_self
):
73 comp
= graph
.add_component(MySink
, 'lel')
76 self
.assertTrue(finalized
)
79 class GenericComponentTestCase(unittest
.TestCase
):
81 def _create_comp(comp_cls
, name
=None):
87 return graph
.add_component(comp_cls
, name
)
90 class MySink(bt2
._UserSinkComponent
):
94 comp
= self
._create
_comp
(MySink
, 'yaes')
95 self
.assertEqual(comp
.name
, 'yaes')
98 class MySink(bt2
._UserSinkComponent
):
103 comp
= graph
.add_component(MySink
, 'lel')
104 self
.assertEqual(comp
.graph
, graph
)
106 def test_class(self
):
107 class MySink(bt2
._UserSinkComponent
):
111 comp
= self
._create
_comp
(MySink
)
112 self
.assertEqual(comp
.component_class
, MySink
)
115 class MySink(bt2
._UserSinkComponent
):
119 comp
= self
._create
_comp
(MySink
)
120 self
.assertIsInstance(comp
.addr
, int)
121 self
.assertNotEqual(comp
.addr
, 0)
This page took 0.043048 seconds and 4 git commands to generate.