def test_make_tree_postaction():
def post_action(t):
t.foo = 'bar'
rd = RAW_DATA[0]
seg_ids = utils.get_initial_neurite_segment_ids(rd)
trees = [utils.make_point_tree(rd, root_id=seg_id, post_action=post_action)
for seg_id in seg_ids]
for t in trees:
nt.ok_(hasattr(t, 'foo') and t.foo == 'bar')
def test_make_point_tree():
rd = RAW_DATA[0]
seg_ids = utils.get_initial_neurite_segment_ids(rd)
trees = [utils.make_point_tree(rd, seg_id) for seg_id in seg_ids]
nt.ok_(len(trees) == len(INIT_IDS[0]))
_check_trees(trees)