def test_segment_lengths(): T = NEURON_TREE lg = [l for l in i_segment_length(T)] nt.assert_equal(lg, [1.0, 1.0, 2.0, 1.0, 2.0, 1.0, 1.0, 2.0, 1.0, 1.0])
def test_get_segment_lengths(self): ref_seglen = [] for t in self.neuron.neurites: ref_seglen.extend(ll for ll in i_segment_length(t)) seglen = self.neuron.get_segment_lengths() nt.assert_equal(len(seglen), 840) nt.assert_true(np.all(seglen == ref_seglen)) seglen = self.neuron.get_segment_lengths(TreeType.all) nt.assert_equal(len(seglen), 840) nt.assert_true(np.all(seglen == ref_seglen))