def test_siblings(self): tree = Tree() n1 = 'n1' n2 = 'n2' n3 = 'n3' tree.add(n1) tree.add(n2) tree.add(n3) assert tree.get_next_sibling(n1) is n2 assert tree.get_next_sibling(n2) is n3 try: tree.get_next_sibling(n3) except IndexError: pass # okay else: raise AssertionError, 'Index should be out of range, not %s' % tree.get_next_sibling(n3) assert tree.get_previous_sibling(n3) is n2 assert tree.get_previous_sibling(n2) is n1 try: tree.get_previous_sibling(n1) except IndexError: pass # okay else: raise AssertionError, 'Index should be out of range, not %s' % tree.get_previous_sibling(n1)