def test_check_create_raw_session_flag(self): raw_session = Path(self.local_session_path_3A).joinpath("raw_session.flag") ephys = Path(self.local_session_path_3A).joinpath("ephys_data_transferred.flag") video = Path(self.local_session_path_3A).joinpath("video_data_transferred.flag") # Check not created misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertFalse(raw_session.exists()) # Create only ephys flag misc.create_ephys_transfer_done_flag(self.local_session_path_3A) # Check not created misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertFalse(raw_session.exists()) ephys.unlink() # Create only video flag misc.create_video_transfer_done_flag(self.local_session_path_3A) # Check not created misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertFalse(raw_session.exists()) video.unlink() # Create ephys and video flag file for completed transfer misc.create_ephys_transfer_done_flag(self.local_session_path_3A) misc.create_video_transfer_done_flag(self.local_session_path_3A) # Check it was created misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertTrue(raw_session.exists()) # Check other flags deleted self.assertFalse(ephys.exists()) self.assertFalse(video.exists()) raw_session.unlink()
def test_create_ephys_transfer_done_flag(self): # Create ephys flag file for completed transfer misc.create_ephys_transfer_done_flag(self.local_session_path_3A) # Check it was created ephys = Path(self.local_session_path_3A).joinpath("ephys_data_transferred.flag") self.assertTrue(ephys.exists()) # Remove it ephys.unlink()
def test_check_create_raw_session_flag(self): from ibllib.tests.fixtures import utils as futils raw_session = Path( self.local_session_path_3A).joinpath("raw_session.flag") ephys = Path( self.local_session_path_3A).joinpath("ephys_data_transferred.flag") video = Path( self.local_session_path_3A).joinpath("video_data_transferred.flag") # Add settings file fpath = self.local_session_path_3A / "raw_behavior_data" / "_iblrig_taskSettings.raw.json" fpath.touch() futils.populate_task_settings( fpath, patch={"PYBPOD_PROTOCOL": "some_ephysChoiceWorld_task"}) "" # Check not created misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertFalse(raw_session.exists()) # Create only ephys flag misc.create_ephys_transfer_done_flag(self.local_session_path_3A) # Check not created misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertFalse(raw_session.exists()) ephys.unlink() # Create only video flag misc.create_video_transfer_done_flag(self.local_session_path_3A) # Check not created misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertFalse(raw_session.exists()) video.unlink() # Create ephys and video flag file for completed transfer misc.create_ephys_transfer_done_flag(self.local_session_path_3A) misc.create_video_transfer_done_flag(self.local_session_path_3A) # Check it was created misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertTrue(raw_session.exists()) # Check other flags deleted self.assertFalse(ephys.exists()) self.assertFalse(video.exists()) raw_session.unlink() # Check if biased session futils.populate_task_settings( fpath, patch={"PYBPOD_PROTOCOL": "some_biasedChoiceWorld_task"}) misc.create_video_transfer_done_flag(self.local_session_path_3A) misc.check_create_raw_session_flag(self.local_session_path_3A) self.assertTrue(raw_session.exists()) self.assertFalse(video.exists()) raw_session.unlink()
def test_check_create_raw_session_flag(self): raw_session = Path(self.session_path_3B).joinpath("raw_session.flag") ephys = Path( self.session_path_3B).joinpath("ephys_data_transferred.flag") video = Path( self.session_path_3B).joinpath("video_data_transferred.flag") # Add settings file fpath = self.session_path_3B / "raw_behavior_data" / "_iblrig_taskSettings.raw.json" fu.populate_task_settings( fpath, patch={"PYBPOD_PROTOCOL": "some_ephysChoiceWorld_task"}) # Check not created misc.check_create_raw_session_flag(self.session_path_3B) self.assertFalse(raw_session.exists()) # Create only ephys flag misc.create_ephys_transfer_done_flag(self.session_path_3B) # Check not created misc.check_create_raw_session_flag(self.session_path_3B) self.assertFalse(raw_session.exists()) ephys.unlink() # Create only video flag misc.create_video_transfer_done_flag(self.session_path_3B) # Check not created misc.check_create_raw_session_flag(self.session_path_3B) self.assertFalse(raw_session.exists()) video.unlink() # Create ephys and video flag file for completed transfer misc.create_ephys_transfer_done_flag(self.session_path_3B) misc.create_video_transfer_done_flag(self.session_path_3B) # Check it was created misc.check_create_raw_session_flag(self.session_path_3B) self.assertTrue(raw_session.exists()) # Check other flags deleted self.assertFalse(ephys.exists()) self.assertFalse(video.exists()) raw_session.unlink() # Check if biased session fu.populate_task_settings( fpath, patch={"PYBPOD_PROTOCOL": "some_biasedChoiceWorld_task"}) misc.create_video_transfer_done_flag(self.session_path_3B) misc.check_create_raw_session_flag(self.session_path_3B) self.assertTrue(raw_session.exists()) self.assertFalse(video.exists()) raw_session.unlink() # Check if training session fu.populate_task_settings( fpath, patch={"PYBPOD_PROTOCOL": "some_trainingChoiceWorld_task"}) misc.create_video_transfer_done_flag(self.session_path_3B) misc.check_create_raw_session_flag(self.session_path_3B) self.assertTrue(raw_session.exists()) self.assertFalse(video.exists()) raw_session.unlink() # Check if habituation session fu.populate_task_settings( fpath, patch={"PYBPOD_PROTOCOL": "some_habituationChoiceWorld_task"}) misc.create_video_transfer_done_flag(self.session_path_3B) misc.check_create_raw_session_flag(self.session_path_3B) self.assertTrue(raw_session.exists()) self.assertFalse(video.exists()) raw_session.unlink()