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_(get_tree_type(test_tree) == tree_types[en_tree]) find_tree_type(test_tree) # tree.type should already exists here, from previous action. nt.ok_(get_tree_type(test_tree) == tree_types[en_tree])
def test_find_tree_type(): for en_tree, test_tree in enumerate(neuron0.neurites): nt.ok_(find_tree_type(test_tree) == tree_types[en_tree])