from neurom.point_neurite import io
from neurom.point_neurite.io.utils import make_neuron
from neurom.point_neurite.point_tree import PointTree
from neurom.point_neurite import bifurcations as bif
from neurom.point_neurite.core import iter_neurites
import math
class MockNeuron(object):
pass
DATA_PATH = './test_data'
SWC_PATH = os.path.join(DATA_PATH, 'swc/')
data = io.load_data(SWC_PATH + 'Neuron.swc')
neuron0 = make_neuron(data)
tree0 = neuron0.neurites[0]
def _make_neuron_tree():
p = [0.0, 0.0, 0.0, 1.0, 1, 1, 2]
T = PointTree(p)
T1 = T.add_child(PointTree([0.0, 1.0, 0.0, 1.0, 1, 1, 2]))
T2 = T1.add_child(PointTree([0.0, 2.0, 0.0, 1.0, 1, 1, 2]))
T3 = T2.add_child(PointTree([0.0, 4.0, 0.0, 2.0, 1, 1, 2]))
T4 = T3.add_child(PointTree([0.0, 5.0, 0.0, 2.0, 1, 1, 2]))
T5 = T4.add_child(PointTree([2.0, 5.0, 0.0, 1.0, 1, 1, 2]))
T6 = T4.add_child(PointTree([0.0, 5.0, 2.0, 1.0, 1, 1, 2]))
T7 = T5.add_child(PointTree([3.0, 5.0, 0.0, 0.75, 1, 1, 2]))
T8 = T7.add_child(PointTree([4.0, 5.0, 0.0, 0.75, 1, 1, 2]))
T9 = T6.add_child(PointTree([0.0, 5.0, 3.0, 0.75, 1, 1, 2]))
from nose import tools as nt
from neurom.point_neurite.io.utils import make_neuron
from neurom import io
from neurom.point_neurite import io as pt_io
from neurom.view import view
from neurom.point_neurite.treefunc import set_tree_type
import os
import numpy as np
import pylab as plt
from neurom.point_neurite.point_tree import PointTree
DATA_PATH = './test_data'
SWC_PATH = os.path.join(DATA_PATH, 'swc/')
pt_raw_data = pt_io.load_data(SWC_PATH + 'Neuron.swc')
pt_neuron = make_neuron(pt_raw_data, set_tree_type)
soma0 = pt_neuron.soma
fst_neuron = io.load_neuron(os.path.join(SWC_PATH, 'Neuron.swc'))
def test_tree():
axes = []
for tree in pt_neuron.neurites:
fig, ax = view.tree(tree)
axes.append(ax)
nt.ok_(axes[0].get_data_ratio() > 1.00 )
nt.ok_(axes[1].get_data_ratio() > 0.80 )
nt.ok_(axes[2].get_data_ratio() > 1.00 )
nt.ok_(axes[3].get_data_ratio() > 0.85 )
