def test_get_section_lengths(self): ref_seclen = [] for t in self.neuron.neurites: ref_seclen.extend(ll for ll in i_section_length(t)) seclen = self.neuron.get_section_lengths() nt.assert_equal(len(seclen), 84) nt.assert_true(np.all(seclen == ref_seclen)) seclen = self.neuron.get_section_lengths(TreeType.all) nt.assert_equal(len(seclen), 84) nt.assert_true(np.all(seclen == ref_seclen))
def test_i_section_length(): T = SIMPLE_TREE nt.assert_equal([l for l in i_section_length(T)], [8.0, 8.0]) T2 = NEURON_TREE nt.ok_([l for l in i_section_length(T2)] == [5.0, 4.0, 4.0])