def test_tree_view_shrinking(self):
handle = SampleTreeHandle()
tree_viewer = TreeView(self.parent, handle)
tree_viewer.add_entry('a')
tree_viewer.expand(0)
self.assertEquals(4, tree_viewer.size())
tree_viewer.collapse(0)
self.assertEquals(1, tree_viewer.size())
self.assertTrue(tree_viewer.get(0).endswith('element a'))
def test_tree_view_multi_expanding(self):
handle = SampleTreeHandle()
tree_viewer = TreeView(self.parent, handle)
tree_viewer.add_entry('a')
tree_viewer.expand(0)
tree_viewer.expand(0)
self.assertEquals(4, tree_viewer.size())
def test_tree_view_expanding(self):
handle = SampleTreeHandle()
tree_viewer = TreeView(self.parent, handle)
tree_viewer.add_entry('a')
tree_viewer.expand(0)
self.assertEquals(4, tree_viewer.size())
self.assertTrue(tree_viewer.get(0).endswith('element a'))
self.assertTrue(tree_viewer.get(1).endswith('element a0'))
self.assertTrue(tree_viewer.get(2).endswith('element a1'))
self.assertTrue(tree_viewer.get(3).endswith('element a2'))
