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)