def test_trunk_origin_elevations():
class Mock(object):
pass
n0 = Mock()
n1 = Mock()
s = make_soma([[0, 0, 0, 4]])
t0 = Section(((1, 0, 0, 2), (2, 1, 1, 2)))
t0.type = NeuriteType.basal_dendrite
t1 = Section(((0, 1, 0, 2), (1, 2, 1, 2)))
t1.type = NeuriteType.basal_dendrite
n0.neurites = [Neurite(t0), Neurite(t1)]
n0.soma = s
t2 = Section(((0, -1, 0, 2), (-1, -2, -1, 2)))
t2.type = NeuriteType.basal_dendrite
n1.neurites = [Neurite(t2)]
n1.soma = s
pop = Population([n0, n1])
nt.eq_(list(_nf.trunk_origin_elevations(pop)), [0.0, np.pi / 2.0, -np.pi / 2.0])
nt.eq_(
list(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.basal_dendrite)),
[0.0, np.pi / 2.0, -np.pi / 2.0],
)
nt.eq_(len(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.axon)), 0)
nt.eq_(len(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.apical_dendrite)), 0)
def test_trunk_origin_elevations():
class Mock(object):
pass
n0 = Mock()
n1 = Mock()
s = make_soma([[0, 0, 0, 4]])
t0 = Section(((1, 0, 0, 2), (2, 1, 1, 2)))
t0.type = NeuriteType.basal_dendrite
t1 = Section(((0, 1, 0, 2), (1, 2, 1, 2)))
t1.type = NeuriteType.basal_dendrite
n0.neurites = [Neurite(t0), Neurite(t1)]
n0.soma = s
t2 = Section(((0, -1, 0, 2), (-1, -2, -1, 2)))
t2.type = NeuriteType.basal_dendrite
n1.neurites = [Neurite(t2)]
n1.soma = s
pop = Population([n0, n1])
nt.eq_(list(_nf.trunk_origin_elevations(pop)),
[0.0, np.pi/2., -np.pi/2.])
nt.eq_(
list(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.basal_dendrite)),
[0.0, np.pi/2., -np.pi/2.])
nt.eq_(len(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.axon)),
0)
nt.eq_(len(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.apical_dendrite)),
0)
def test_trunk_elevation_zero_norm_vector_raises():
_nf.trunk_origin_elevations(SWC_NRN)
def test_trunk_elevation_zero_norm_vector_raises():
_nf.trunk_origin_elevations(NRN)