def test_get_included_trials(self): # TRAINING SESSIONS it = training_trials.IncludedTrials(self.training_lt5['path']).extract( settings={'IBLRIG_VERSION_TAG': '4.9.9'})[0] self.assertTrue(isinstance(it, np.ndarray)) # -- version >= 5.0.0 it = training_trials.IncludedTrials( self.training_ge5['path']).extract()[0] self.assertTrue(isinstance(it, np.ndarray)) # BIASED SESSIONS it = biased_trials.IncludedTrials(self.biased_lt5['path']).extract( settings={'IBLRIG_VERSION_TAG': '4.9.9'})[0] self.assertTrue(isinstance(it, np.ndarray)) # -- version >= 5.0.0 it = biased_trials.IncludedTrials(self.biased_ge5['path']).extract()[0] self.assertTrue(isinstance(it, np.ndarray))
def test_get_included_trials_ge5(self): # TRAINING SESSIONS it = training_trials.IncludedTrials( self.training_ge5['path']).extract()[0] self.assertTrue(isinstance(it, np.ndarray)) # BIASED SESSIONS it = biased_trials.IncludedTrials(self.biased_ge5['path']).extract()[0] self.assertTrue(isinstance(it, np.ndarray))