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)