def test_get_tree_type():
for en_tree, test_tree in enumerate(neuron0.neurites):
if hasattr(test_tree, 'type'):
del test_tree.type
# tree.type should be computed here.
nt.ok_(mtr.get_tree_type(test_tree) == tree_types[en_tree])
mtr.find_tree_type(test_tree)
# tree.type should already exists here, from previous action.
nt.ok_(mtr.get_tree_type(test_tree) == tree_types[en_tree])
def neurite_type(neurite):
'''Get the neurite type of a neurite tree'''
if is_new_style(neurite):
return neurite.type
else:
return find_tree_type(neurite)
def test_find_tree_type():
for en_tree, test_tree in enumerate(neuron0.neurites):
nt.ok_(mtr.find_tree_type(test_tree) == tree_types[en_tree])
