Python set_tree_type示例

示例#1

文件： test_dendrogram.py 项目： juanchopanza/NeuroM

from nose import tools as nt
from neurom.point_neurite.point_tree import PointTree
from neurom.point_neurite.core import Neuron
from neurom.core import make_soma
from neurom.core import NeuriteType
from neurom.point_neurite.treefunc import set_tree_type
import neurom.point_neurite.dendrogram as dm


TREE = PointTree(np.array([0., 0., 0., 10., 4., 0., 0.]))
TREE.add_child(PointTree(np.array([3., 3., 3., 9., 4., 0., 0.])))

TREE.children[0].add_child(PointTree(np.array([10., 10., 10., 5., 4., 0., 0.])))
TREE.children[0].add_child(PointTree(np.array([-10., -10., -10., 7., 4., 0., 0.])))

set_tree_type(TREE)

SOMA = make_soma(np.array([[0., 0., 0., 1., 1., 1., -1.]]))
NEURON = Neuron(SOMA, [TREE, TREE, TREE])

OLD_OFFS = [1.2, -1.2]
NEW_OFFS = [2.3, -2.3]
SPACING = (40., 0.)

def test_n_rectangles_tree():

    nt.assert_equal(dm._n_rectangles(TREE), 5)


def test_n_rectangles_neuron():

示例#2

文件： test_treefunc.py 项目： juanchopanza/NeuroM

def test_set_tree_type():
    for en_tree, test_tree in enumerate(neuron0.neurites):
        mtr.set_tree_type(test_tree)
        nt.ok_(test_tree.type == tree_types[en_tree])