def test_trunk_origin_azimuths():
n0 = MockNeuron()
n1 = MockNeuron()
n2 = MockNeuron()
n3 = MockNeuron()
n4 = MockNeuron()
n5 = MockNeuron()
t = PointTree((0, 0, 0, 2))
t.type = NeuriteType.basal_dendrite
n0.neurites = [t]
n1.neurites = [t]
n2.neurites = [t]
n3.neurites = [t]
n4.neurites = [t]
n5.neurites = [t]
pop = [n0, n1, n2, n3, n4, n5]
s0 = make_soma([[0, 0, 1, 4]])
s1 = make_soma([[0, 0, -1, 4]])
s2 = make_soma([[0, 0, 0, 4]])
s3 = make_soma([[-1, 0, -1, 4]])
s4 = make_soma([[-1, 0, 0, 4]])
s5 = make_soma([[1, 0, 0, 4]])
pop[0].soma = s0
pop[1].soma = s1
pop[2].soma = s2
pop[3].soma = s3
pop[4].soma = s4
pop[5].soma = s5
nt.assert_true(np.all(get_feat('trunk_origin_azimuths', pop) ==
[-np.pi/2., np.pi/2., 0.0, np.pi/4., 0.0, np.pi]))
nt.eq_(len(get_feat('trunk_origin_azimuths', pop, neurite_type=NeuriteType.axon)), 0)
def test_trunk_origin_elevations():
n0 = MockNeuron()
n1 = MockNeuron()
s = make_soma([[0, 0, 0, 4]])
t0 = PointTree((1, 0, 0, 2))
t0.type = NeuriteType.basal_dendrite
t1 = PointTree((0, 1, 0, 2))
t1.type = NeuriteType.basal_dendrite
n0.neurites = [t0, t1]
n0.soma = s
t2 = PointTree((0, -1, 0, 2))
t2.type = NeuriteType.basal_dendrite
n1.neurites = [t2]
n1.soma = s
pop = [n0, n1]
nt.assert_true(np.all(get_feat('trunk_origin_elevations', pop) ==
[0.0, np.pi/2., -np.pi/2.]))
nt.eq_(len(get_feat('trunk_origin_elevations', pop, neurite_type=NeuriteType.axon)), 0)
