Python set_tree_type示例

示例#1

文件： test_morphtree.py 项目： wvangeit/NeuroM

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

示例#2

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

import numpy as np
from nose import tools as nt
from neurom.core.tree import Tree
from neurom.core.neuron import Neuron, make_soma
from neurom.core.types import TreeType
from neurom.analysis.morphtree import set_tree_type
import neurom.analysis.dendrogram as dm

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

TREE.children[0].add_child(Tree(np.array([10., 10., 10., 5., 4., 0., 0.])))
TREE.children[0].add_child(Tree(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():