def test_i_section_radial_dists():
T1 = SIMPLE_TREE
T2 = NEURON_TREE
p0 = [0.0, 0.0, 0.0]
nt.assert_equal([d for d in i_section_radial_dist(T1)],
[8.0, 8.0])
nt.assert_equal([d for d in i_section_radial_dist(T1, p0)],
[8.0, 8.0])
nt.assert_equal([d for d in i_section_radial_dist(T1, use_start_point=True)],
[0.0, 0.0])
nt.assert_equal([d for d in i_section_radial_dist(T1, p0, use_start_point=True)],
[0.0, 0.0])
nt.assert_true(np.allclose([d for d in i_section_radial_dist(T2)],
[5.0, 6.4031, 6.7082]))
nt.assert_true(np.allclose([d for d in i_section_radial_dist(T2, p0)],
[5.0, 6.4031, 6.7082]))
nt.assert_equal([d for d in i_section_radial_dist(T2, use_start_point=True)],
[0.0, 5.0, 5.0])
nt.assert_equal([d for d in i_section_radial_dist(T2, p0, use_start_point=True)],
[0.0, 5.0, 5.0])
def test_get_section_radial_distances_endpoint(self):
ref_sec_rad_dist_start = []
for t in self.neuron.neurites:
ref_sec_rad_dist_start.extend(
ll for ll in i_section_radial_dist(t, use_start_point=True))
ref_sec_rad_dist = []
for t in self.neuron.neurites:
ref_sec_rad_dist.extend(ll for ll in i_section_radial_dist(t))
rad_dists = self.neuron.get_section_radial_distances()
nt.assert_true(ref_sec_rad_dist != ref_sec_rad_dist_start)
nt.assert_equal(len(rad_dists), 84)
nt.assert_true(np.all(rad_dists == ref_sec_rad_dist))
def test_get_section_radial_distances_endpoint(self):
ref_sec_rad_dist_start = []
for t in self.neuron.neurites:
ref_sec_rad_dist_start.extend(
ll for ll in i_section_radial_dist(t, use_start_point=True))
ref_sec_rad_dist = []
for t in self.neuron.neurites:
ref_sec_rad_dist.extend(ll for ll in i_section_radial_dist(t))
rad_dists = self.neuron.get_section_radial_distances()
nt.assert_true(ref_sec_rad_dist != ref_sec_rad_dist_start)
nt.assert_equal(len(rad_dists), 84)
nt.assert_true(np.all(rad_dists == ref_sec_rad_dist))
def get_section_radial_distances(self, origin=None, use_start_point=False,
neurite_type=TreeType.all):
'''Get an iterable containing section radial distances to origin of\
all neurites of a given type
Parameters:
origin: Point wrt which radial dirtance is calulated\
(default tree root)
use_start_point: if true, use the section's first point,\
otherwise use the end-point (default False)
neurite_type: Type of neurites to be considered (default all)
'''
return self.neurite_loop(lambda t: i_section_radial_dist(t, origin,
use_start_point),
neurite_type=neurite_type)
def get_section_radial_distances(self,
origin=None,
use_start_point=False,
neurite_type=TreeType.all):
'''Get an iterable containing section radial distances to origin of\
all neurites of a given type
Parameters:
origin: Point wrt which radial dirtance is calulated\
(default tree root)
use_start_point: if true, use the section's first point,\
otherwise use the end-point (default False)
neurite_type: Type of neurites to be considered (default all)
'''
return self.neurite_loop(
lambda t: i_section_radial_dist(t, origin, use_start_point),
neurite_type=neurite_type)
