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])