def print_tree(cur, start_node=None, indent=' ', _level=0):
"""
Print tree to stdout.
:param start_node: Starting point for tree output.
If ``None``, start at root node.
:type start_node: int, Node, NodaData or None
:param str indent: String to print per level (default: ' ')
"""
if start_node is None:
start_node = get_root_node(cur)
print('{}{}'.format(indent*_level, start_node)) # noqa
for child in list(get_children(cur, start_node)):
print_tree(cur, child, indent=indent, _level=_level+1)
def print_tree(cur, start_node=None, indent=' ', _level=0):
"""
Print tree to stdout.
:param start_node: Starting point for tree output.
If ``None``, start at root node.
:type start_node: int, Node, NodaData or None
:param str indent: String to print per level (default: ' ')
"""
if start_node is None:
start_node = get_root_node(cur)
print('{}{}'.format(indent*_level, start_node)) # noqa
for child in list(get_children(cur, start_node)):
print_tree(cur, child, indent=indent, _level=_level+1)
def test_get_root_node(cur, root, nd1, nd2, nd2_1, nd2_1_1, nd3):
root = get_root_node(cur)
assert root.parent is None
def test_get_root_node_non_existing(cur):
with pytest.raises(exceptions.NoRootNode):
get_root_node(cur)
def test_get_root_node_non_existing(cur):
with pytest.raises(ValueError):
get_root_node(cur)
def test_get_root_node(cur, root, nd1, nd2, nd2_1, nd2_1_1, nd3):
root = get_root_node(cur)
assert root.parent is None
def test_get_root_node_non_existing(cur):
with pytest.raises(exceptions.NoRootNode):
get_root_node(cur)
def test_get_root_node_non_existing(cur):
with pytest.raises(ValueError):
get_root_node(cur)