def test_expansion_signs(self):
handle = SampleTreeHandle()
tree_viewer = TreeView(self.parent, handle)
tree_viewer.add_entry('a')
self.assertEquals('+ element a', tree_viewer.get(0))
tree_viewer.expand(0)
self.assertEquals('- element a', tree_viewer.get(0))
tree_viewer.collapse(0)
self.assertEquals('+ element a', tree_viewer.get(0))
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'))
