def test_simple_usage(self):
# Given
cs = tvtk.ConeSource()
ef = tvtk.ElevationFilter(input_connection=cs.output_port)
# When
p = PipelineBrowser(root_object=[ef])
# Then
self.assertEqual(len(p._root.children), 1)
kids = list(p._root.children)
self.assertEqual(kids[0].name, 'ElevationFilter')
gk = list(kids[0].children)
self.assertEqual(len(gk), 1)
self.assertEqual(gk[0].name, 'ConeSource')
ggk = list(gk[0].children)
self.assertEqual(len(ggk), 0)
# Check if the default traits_view returns correctly.
p.default_traits_view()
# When
# Check if editing a selected object fires a ui change.
self.count = 0
def callback():
self.count += 1
p.on_trait_change(callback, 'object_edited')
p._on_select(gk[0])
cs.height = 2.0
# Then
self.assertTrue(self.count > 0)
def _browser_default(self):
b = PipelineBrowser()
b.on_trait_change(self._fire_data_changed, 'object_edited')
return b
