flat-tree/test/test_iterator.py

24 lines
716 B
Python
Raw Normal View History

def test_iter_from_leaf(FlatTreeIterator):
tree_iter = FlatTreeIterator()
assert tree_iter.index == 0
assert tree_iter.parent() == 1
assert tree_iter.parent() == 3
assert tree_iter.parent() == 7
assert tree_iter.right_child() == 11
assert tree_iter.left_child() == 9
assert tree_iter.next() == 13
assert tree_iter.left_span() == 12
def test_iter_not_from_leaf(FlatTreeIterator):
tree_iter = FlatTreeIterator(index=1)
assert tree_iter.index == 1
assert tree_iter.parent() == 3
assert tree_iter.parent() == 7
assert tree_iter.right_child() == 11
assert tree_iter.left_child() == 9
assert tree_iter.next() == 13
assert tree_iter.left_span() == 12