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'))