bt2: clean available `bt2` package names

[babeltrace.git] / tests / bindings / python / bt2 / test_component_class.py

diff --git a/tests/bindings/python/bt2/test_component_class.py b/tests/bindings/python/bt2/test_component_class.py
index 3e175c83a02a3a6afeaec3e2da2a2cc585f3defc..909b2d642562c4d8b17ec98013529a1da3943a03 100644 (file)
--- a/tests/bindings/python/bt2/test_component_class.py
+++ b/tests/bindings/python/bt2/test_component_class.py
@@ -382,7 +382,7 @@ class ComponentClassTestCase(unittest.TestCase):
         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