def test_electrode_group_factory_create_ElectrodeGroup_successfully(self): mock_probe = Mock(spec=Probe) mock_device = Mock(spec=Device) mock_fl_electrode_group_1 = Mock(spec=FlElectrodeGroup) mock_fl_electrode_group_1.name = '0' mock_fl_electrode_group_1.description = 'ElectrodeGroup 1' mock_fl_electrode_group_1.location = 'mPFC' mock_fl_electrode_group_1.device = mock_probe mock_fl_electrode_group_2 = Mock(spec=FlElectrodeGroup) mock_fl_electrode_group_2.name = '1' mock_fl_electrode_group_2.description = 'ElectrodeGroup 2' mock_fl_electrode_group_2.location = 'mPFC' mock_fl_electrode_group_2.device = mock_device electrode_group_1 = ElectrodeGroupFactory.create_electrode_group( mock_fl_electrode_group_1) electrode_group_2 = ElectrodeGroupFactory.create_electrode_group( mock_fl_electrode_group_2) self.assertIsNotNone(electrode_group_1) self.assertIsNotNone(electrode_group_2) self.assertEqual(electrode_group_1.name, "0") self.assertEqual(electrode_group_1.description, 'ElectrodeGroup 1') self.assertEqual(electrode_group_1.location, 'mPFC') self.assertEqual(electrode_group_1.device, mock_probe) self.assertEqual(electrode_group_2.name, '1') self.assertEqual(electrode_group_2.description, 'ElectrodeGroup 2') self.assertEqual(electrode_group_2.location, 'mPFC') self.assertEqual(electrode_group_2.device, mock_device)
def test_electrode_group_factory_failed_creating_ElectrodeGroup_due_to_lack_of_FlElectrodeGroup_attr( self): mock_fl_electrode_group_1 = Mock(spec=FlElectrodeGroup) mock_fl_electrode_group_1.name = 'ElectrodeGroup 1' mock_fl_electrode_group_1.description = 'sample desciption 1' mock_fl_electrode_group_1.location = 'sample location 1' mock_fl_electrode_group_1.device = None ElectrodeGroupFactory.create_electrode_group(mock_fl_electrode_group_1)
def test_electrode_group_factory_create_NwbElectrodeGroup_successfully( self): mock_probe = Mock(spec=Probe) mock_device = Mock(spec=Device) mock_fl_nwb_electrode_group_1 = Mock(spec=FlNwbElectrodeGroup) mock_fl_nwb_electrode_group_1.name = '0' mock_fl_nwb_electrode_group_1.description = 'ElectrodeGroup 1' mock_fl_nwb_electrode_group_1.location = 'mPFC' mock_fl_nwb_electrode_group_1.device = mock_probe mock_fl_nwb_electrode_group_1.targeted_location = 'Sample location' mock_fl_nwb_electrode_group_1.targeted_x = 0.0 mock_fl_nwb_electrode_group_1.targeted_y = 0.0 mock_fl_nwb_electrode_group_1.targeted_z = 0.0 mock_fl_nwb_electrode_group_1.units = 'um' mock_fl_nwb_electrode_group_2 = Mock(spec=FlNwbElectrodeGroup) mock_fl_nwb_electrode_group_2.name = '1' mock_fl_nwb_electrode_group_2.description = 'ElectrodeGroup 2' mock_fl_nwb_electrode_group_2.location = 'mPFC' mock_fl_nwb_electrode_group_2.device = mock_device mock_fl_nwb_electrode_group_2.targeted_location = 'Sample location' mock_fl_nwb_electrode_group_2.targeted_x = 0.0 mock_fl_nwb_electrode_group_2.targeted_y = 0.0 mock_fl_nwb_electrode_group_2.targeted_z = 0.0 mock_fl_nwb_electrode_group_2.units = 'mm' nwb_electrode_group_1 = ElectrodeGroupFactory.create_nwb_electrode_group( mock_fl_nwb_electrode_group_1) nwb_electrode_group_2 = ElectrodeGroupFactory.create_nwb_electrode_group( mock_fl_nwb_electrode_group_2) self.assertIsNotNone(nwb_electrode_group_1) self.assertIsNotNone(nwb_electrode_group_2) self.assertEqual(nwb_electrode_group_1.name, "0") self.assertEqual(nwb_electrode_group_1.description, 'ElectrodeGroup 1') self.assertEqual(nwb_electrode_group_1.location, 'mPFC') self.assertEqual(nwb_electrode_group_1.device, mock_probe) self.assertEqual(nwb_electrode_group_1.targeted_location, 'Sample location') self.assertEqual(nwb_electrode_group_1.targeted_x, 0.0) self.assertEqual(nwb_electrode_group_1.targeted_y, 0.0) self.assertEqual(nwb_electrode_group_1.targeted_z, 0.0) self.assertEqual(nwb_electrode_group_1.units, 'um') self.assertEqual(nwb_electrode_group_2.name, '1') self.assertEqual(nwb_electrode_group_2.description, 'ElectrodeGroup 2') self.assertEqual(nwb_electrode_group_2.location, 'mPFC') self.assertEqual(nwb_electrode_group_2.device, mock_device) self.assertEqual(nwb_electrode_group_2.targeted_location, 'Sample location') self.assertEqual(nwb_electrode_group_2.targeted_x, 0.0) self.assertEqual(nwb_electrode_group_2.targeted_y, 0.0) self.assertEqual(nwb_electrode_group_2.targeted_z, 0.0) self.assertEqual(nwb_electrode_group_2.units, 'mm')
def test_electrode_group_factory_failed_creating_NwbElectrodeGroup_due_to_lack_of_FlNwbElectrodeGroup_attr( self): mock_fl_nwb_electrode_group_1 = Mock(spec=FlNwbElectrodeGroup) mock_fl_nwb_electrode_group_1.name = 'ElectrodeGroup 1' mock_fl_nwb_electrode_group_1.description = 'sample desciption 1' mock_fl_nwb_electrode_group_1.location = 'sample location 1' mock_fl_nwb_electrode_group_1.device = None mock_fl_nwb_electrode_group_1.targeted_location = None mock_fl_nwb_electrode_group_1.targeted_x = None mock_fl_nwb_electrode_group_1.targeted_y = None mock_fl_nwb_electrode_group_1.targeted_z = None mock_fl_nwb_electrode_group_1.units = None ElectrodeGroupFactory.create_nwb_electrode_group( mock_fl_nwb_electrode_group_1)
class ElectrodeGroupOriginator: def __init__(self, metadata): self.fl_nwb_electrode_group_manager = FlNwbElectrodeGroupManager( metadata['electrode groups']) self.electrode_group_creator = ElectrodeGroupFactory() self.electrode_group_injector = ElectrodeGroupInjector() def make(self, nwb_content, probes, electrode_groups_valid_map): logger.info('ElectrodeGroups: Building') fl_nwb_electrode_groups = self.fl_nwb_electrode_group_manager.get_fl_nwb_electrode_groups( probes=probes, electrode_groups_valid_map=electrode_groups_valid_map) logger.info('ElectrodeGroups: Creating') nwb_electrode_groups = [ self.electrode_group_creator.create_nwb_electrode_group( nwb_electrode_group) for nwb_electrode_group in fl_nwb_electrode_groups ] logger.info('ElectrodeGroups: Injecting into NWB') self.electrode_group_injector.inject_all_electrode_groups( nwb_content, nwb_electrode_groups) return nwb_electrode_groups
def test_electrode_group_factory_failed_creating_NwbElectrodeGroup_due_to_lack_of_FLNwbElectrodeGroup( self): ElectrodeGroupFactory.create_nwb_electrode_group(None)
def __init__(self, metadata): self.fl_nwb_electrode_group_manager = FlNwbElectrodeGroupManager( metadata['electrode groups']) self.electrode_group_creator = ElectrodeGroupFactory() self.electrode_group_injector = ElectrodeGroupInjector()