Example #1
0
    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)