Commit | Line | Data |
---|---|---|
c4239792 | 1 | from bt2 import value |
811644b8 PP |
2 | import unittest |
3 | import copy | |
4 | import bt2 | |
5 | ||
6 | ||
976c241d | 7 | @unittest.skip("this is broken") |
811644b8 PP |
8 | class UserComponentTestCase(unittest.TestCase): |
9 | @staticmethod | |
10 | def _create_comp(comp_cls, name=None): | |
11 | graph = bt2.Graph() | |
12 | ||
13 | if name is None: | |
14 | name = 'comp' | |
15 | ||
16 | return graph.add_component(comp_cls, name) | |
17 | ||
18 | def test_name(self): | |
19 | class MySink(bt2._UserSinkComponent): | |
20 | def __init__(comp_self, params): | |
21 | self.assertEqual(comp_self.name, 'yaes') | |
22 | ||
23 | def _consume(self): | |
24 | pass | |
25 | ||
26 | comp = self._create_comp(MySink, 'yaes') | |
27 | ||
28 | def test_graph(self): | |
29 | class MySink(bt2._UserSinkComponent): | |
30 | def __init__(comp_self, params): | |
31 | nonlocal graph | |
32 | self.assertEqual(comp_self.graph, graph) | |
33 | ||
34 | def _consume(self): | |
35 | pass | |
36 | ||
37 | graph = bt2.Graph() | |
38 | comp = graph.add_component(MySink, 'lel') | |
39 | del graph | |
40 | ||
41 | def test_class(self): | |
42 | class MySink(bt2._UserSinkComponent): | |
43 | def __init__(comp_self, params): | |
44 | self.assertEqual(comp_self.component_class, MySink) | |
45 | ||
46 | def _consume(self): | |
47 | pass | |
48 | ||
49 | self._create_comp(MySink) | |
50 | ||
51 | def test_addr(self): | |
52 | class MySink(bt2._UserSinkComponent): | |
53 | def __init__(comp_self, params): | |
54 | self.assertIsInstance(comp_self.addr, int) | |
55 | self.assertNotEqual(comp_self.addr, 0) | |
56 | ||
57 | def _consume(self): | |
58 | pass | |
59 | ||
60 | self._create_comp(MySink) | |
61 | ||
62 | def test_finalize(self): | |
63 | finalized = False | |
64 | ||
65 | class MySink(bt2._UserSinkComponent): | |
66 | def _consume(self): | |
67 | pass | |
68 | ||
69 | def _finalize(comp_self): | |
70 | nonlocal finalized | |
71 | finalized = True | |
72 | ||
73 | graph = bt2.Graph() | |
74 | comp = graph.add_component(MySink, 'lel') | |
75 | del graph | |
76 | del comp | |
77 | self.assertTrue(finalized) | |
78 | ||
79 | ||
976c241d | 80 | @unittest.skip("this is broken") |
811644b8 PP |
81 | class GenericComponentTestCase(unittest.TestCase): |
82 | @staticmethod | |
83 | def _create_comp(comp_cls, name=None): | |
84 | graph = bt2.Graph() | |
85 | ||
86 | if name is None: | |
87 | name = 'comp' | |
88 | ||
89 | return graph.add_component(comp_cls, name) | |
90 | ||
91 | def test_name(self): | |
92 | class MySink(bt2._UserSinkComponent): | |
93 | def _consume(self): | |
94 | pass | |
95 | ||
96 | comp = self._create_comp(MySink, 'yaes') | |
97 | self.assertEqual(comp.name, 'yaes') | |
98 | ||
99 | def test_graph(self): | |
100 | class MySink(bt2._UserSinkComponent): | |
101 | def _consume(self): | |
102 | pass | |
103 | ||
104 | graph = bt2.Graph() | |
105 | comp = graph.add_component(MySink, 'lel') | |
106 | self.assertEqual(comp.graph, graph) | |
107 | ||
108 | def test_class(self): | |
109 | class MySink(bt2._UserSinkComponent): | |
110 | def _consume(self): | |
111 | pass | |
112 | ||
113 | comp = self._create_comp(MySink) | |
114 | self.assertEqual(comp.component_class, MySink) | |
115 | ||
116 | def test_addr(self): | |
117 | class MySink(bt2._UserSinkComponent): | |
118 | def _consume(self): | |
119 | pass | |
120 | ||
121 | comp = self._create_comp(MySink) | |
122 | self.assertIsInstance(comp.addr, int) | |
123 | self.assertNotEqual(comp.addr, 0) |