24 lines
716 B
Python
24 lines
716 B
Python
|
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
|