def test_iter_wont_raise_recursion_error():
"this test setup used to raise a RecursiveErrror due to list.insert method"
parent = SpedNode('parent')
son = SpedNode('son', children=[SpedNode('grandchild')])
assert parent.count() == 1
assert son.count() == 2
parent.insert(son)
assert parent.count() == 3
def test_length_depth_one():
# this node have 3 children with zero children each
node = SpedNode('', [
SpedNode('', []),
SpedNode('', []),
SpedNode('', []),
])
assert node.count() == 4 # 1 + 3
def test_length_depth_two():
# this node has child that has two children.
node = SpedNode('', [
SpedNode('', [
SpedNode('', []),
SpedNode('', []),
])
])
assert node.count() == 4
def test_length_flat():
node = SpedNode('', [])
assert node.count() == 1