def test_trunk_origin_elevations():
n0 = load_neuron(StringIO(u"""
1 1 0 0 0 4 -1
2 3 1 0 0 2 1
3 3 2 1 1 2 2
4 3 0 1 0 2 1
5 3 1 2 1 2 4
"""), reader='swc')
n1 = load_neuron(StringIO(u"""
1 1 0 0 0 4 -1
2 3 0 -1 0 2 1
3 3 -1 -2 -1 2 2
"""), reader='swc')
pop = Population([n0, n1])
assert_array_equal(_nf.trunk_origin_elevations(pop),
[0.0, np.pi/2., -np.pi/2.])
assert_array_equal(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.basal_dendrite),
[0.0, np.pi/2., -np.pi/2.])
assert_array_equal(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.axon),
[])
assert_array_equal(_nf.trunk_origin_elevations(pop, neurite_type=NeuriteType.apical_dendrite),
[])
def test_trunk_elevation_zero_norm_vector_raises():
_nf.trunk_origin_elevations(SWC_NRN)
def test_trunk_elevation_zero_norm_vector_raises():
with pytest.raises(Exception):
_nf.trunk_origin_elevations(SWC_NRN)
