def n_segments(tree):
"""
Return number of segments in tree
"""
return sum(1 for _ in ptr.isegment(tree))
def n_segments(tree):
"""
Return number of segments in tree
"""
return sum(1 for _ in ptr.isegment(tree))
