def test_load_neuron_disconnected_components():
filepath = DISCONNECTED_POINTS_FILE
trees = utils.load_trees(filepath)
nt.eq_(len(trees), 8)
# tree ID - number of points map
ref_ids_pts = {4: 1, 215: 1, 426: 1, 637: 1, 6: 209, 217: 209, 428: 209, 639: 209}
ids_pts = {}
for t in trees:
ids_pts[t.value[COLS.ID]] = pts.count(t)
nt.eq_(ref_ids_pts, ids_pts)
def test_load_neuron_disconnected_components():
filepath = os.path.join(SWC_PATH, 'Neuron_disconnected_components.swc')
trees = utils.load_trees(filepath)
nt.eq_(len(trees), 8)
# tree ID - number of points map
ref_ids_pts = {4: 1, 215: 1, 426: 1, 637: 1, 6: 209, 217: 209, 428: 209, 639: 209}
ids_pts = {}
for t in trees:
ids_pts[t.value[COLS.ID]] = pts.count(t)
nt.eq_(ref_ids_pts, ids_pts)
def _check_count(obj, n):
nt.eq_(pts.count(obj), n)
