class TestDefault(unittest.TestCase):
fe = FeatureExtractor()
for feature in default_features:
fe.register_features([feature])
data = Data(test_morphology_large())
feature_extraction_run = fe.extract(data)
results = feature_extraction_run.results
pass
def test_get_node_by_types_axon(self):
morphology = test_morphology_large()
expected_nodes = [
test_node(id=12,
type=AXON,
x=900,
y=600,
z=30,
radius=3,
parent_node_id=1),
test_node(id=13,
type=AXON,
x=930,
y=630,
z=40,
radius=3,
parent_node_id=12),
test_node(id=14,
type=AXON,
x=960,
y=660,
z=50,
radius=3,
parent_node_id=13),
test_node(id=15,
type=AXON,
x=990,
y=690,
z=60,
radius=3,
parent_node_id=14),
test_node(id=16,
type=AXON,
x=1020,
y=720,
z=70,
radius=3,
parent_node_id=15),
test_node(id=17,
type=AXON,
x=1050,
y=750,
z=80,
radius=3,
parent_node_id=16)
]
nodes = morphology.get_node_by_types([AXON])
self.assertEqual(expected_nodes, nodes)
def test_get_node_by_types_apical(self):
morphology = test_morphology_large()
expected_nodes = [
test_node(id=6,
type=APICAL_DENDRITE,
x=600,
y=300,
z=20,
radius=3,
parent_node_id=1),
test_node(id=7,
type=APICAL_DENDRITE,
x=630,
y=330,
z=30,
radius=3,
parent_node_id=6),
test_node(id=8,
type=APICAL_DENDRITE,
x=660,
y=360,
z=40,
radius=3,
parent_node_id=7),
test_node(id=9,
type=APICAL_DENDRITE,
x=690,
y=390,
z=50,
radius=3,
parent_node_id=8),
test_node(id=10,
type=APICAL_DENDRITE,
x=710,
y=420,
z=60,
radius=3,
parent_node_id=9),
test_node(id=11,
type=APICAL_DENDRITE,
x=740,
y=450,
z=70,
radius=3,
parent_node_id=10)
]
nodes = morphology.get_node_by_types([APICAL_DENDRITE])
self.assertEqual(expected_nodes, nodes)
def test_get_node_by_types_basal(self):
morphology = test_morphology_large()
expected_nodes = [
test_node(id=2,
type=BASAL_DENDRITE,
x=400,
y=600,
z=10,
radius=3,
parent_node_id=1),
test_node(id=3,
type=BASAL_DENDRITE,
x=430,
y=630,
z=20,
radius=3,
parent_node_id=2),
test_node(id=4,
type=BASAL_DENDRITE,
x=460,
y=660,
z=30,
radius=3,
parent_node_id=3),
test_node(id=5,
type=BASAL_DENDRITE,
x=490,
y=690,
z=40,
radius=3,
parent_node_id=4)
]
nodes = morphology.get_node_by_types([BASAL_DENDRITE])
self.assertEqual(expected_nodes, nodes)
def test_get_segment_list(self):
morphology = test_morphology_large()
segments = morphology.get_segment_list()
expected_segments = [[
test_node(id=2,
type=BASAL_DENDRITE,
x=400,
y=600,
z=10,
radius=3,
parent_node_id=1),
test_node(id=3,
type=BASAL_DENDRITE,
x=430,
y=630,
z=20,
radius=3,
parent_node_id=2),
test_node(id=4,
type=BASAL_DENDRITE,
x=460,
y=660,
z=30,
radius=3,
parent_node_id=3),
test_node(id=5,
type=BASAL_DENDRITE,
x=490,
y=690,
z=40,
radius=3,
parent_node_id=4)
],
[
test_node(id=6,
type=APICAL_DENDRITE,
x=600,
y=300,
z=20,
radius=3,
parent_node_id=1),
test_node(id=7,
type=APICAL_DENDRITE,
x=630,
y=330,
z=30,
radius=3,
parent_node_id=6),
test_node(id=8,
type=APICAL_DENDRITE,
x=660,
y=360,
z=40,
radius=3,
parent_node_id=7),
test_node(id=9,
type=APICAL_DENDRITE,
x=690,
y=390,
z=50,
radius=3,
parent_node_id=8),
test_node(id=10,
type=APICAL_DENDRITE,
x=710,
y=420,
z=60,
radius=3,
parent_node_id=9),
test_node(id=11,
type=APICAL_DENDRITE,
x=740,
y=450,
z=70,
radius=3,
parent_node_id=10)
],
[
test_node(id=12,
type=AXON,
x=900,
y=600,
z=30,
radius=3,
parent_node_id=1),
test_node(id=13,
type=AXON,
x=930,
y=630,
z=40,
radius=3,
parent_node_id=12),
test_node(id=14,
type=AXON,
x=960,
y=660,
z=50,
radius=3,
parent_node_id=13),
test_node(id=15,
type=AXON,
x=990,
y=690,
z=60,
radius=3,
parent_node_id=14),
test_node(id=16,
type=AXON,
x=1020,
y=720,
z=70,
radius=3,
parent_node_id=15),
test_node(id=17,
type=AXON,
x=1050,
y=750,
z=80,
radius=3,
parent_node_id=16)
]]
self.assertEqual(expected_segments, segments)
def test_get_max_id(self):
morphology = test_morphology_large()
expected_id = 17
max_id = morphology.get_max_id()
self.assertEqual(expected_id, max_id)