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)