Merge pull request #302 from CAMBI-tech/reshaper_testing

Adds tests to rehsapers

Author: tab-cmd

bcipy/helpers/stimuli.py

@@ -221,7 +221,7 @@ def extract_trials(
 
                 try:
                     new_trials.append(inquiries[:, inquiry_idx, start:end])
-                except IndexError:
+                except IndexError:  # pragma: no cover
                     raise BciPyCoreException(
                         f'InquiryReshaper.extract_trials: index out of bounds. \n'
                         f'Inquiry: [{inquiry_idx}] from {start}:{end}. init_time: {time}, '

bcipy/helpers/tests/test_stimuli.py

@@ -713,6 +713,22 @@ def test_best_case_inquiry_gen(self):
         self.assertEqual([1] + ([0.2] * n), times[0])
         self.assertEqual(['red'] + (['white'] * n), colors[0])
 
+    def test_best_case_inquiry_gen_invalid_alp(self):
+        """Test best_case_rsvp_inq_gen throws error when passed invalid alp shape"""
+        alp = ['a', 'b', 'c', 'd']
+        session_stimuli = [0.1, 0.1, 0.1, 0.2, 0.2, 0.1, 0.2]
+        stim_length = 5
+        with self.assertRaises(BciPyCoreException, msg='Missing information about the alphabet.'):
+            best_case_rsvp_inq_gen(
+                alp=alp,
+                session_stimuli=session_stimuli,
+                timing=[1, 0.2],
+                color=['red', 'white'],
+                stim_number=1,
+                stim_length=stim_length,
+                is_txt=True
+            )
+
     def test_best_case_inquiry_gen_with_inq_constants(self):
         """Test best_case_rsvp_inq_gen with inquiry constants"""
 
@@ -854,6 +870,22 @@ def test_trial_reshaper(self):
         self.assertTrue(np.all(labels == [1, 0, 0]))
         self.assertTrue(reshaped_trials.shape == expected_shape)
 
+    def test_trial_reshaper_with_no_channel_map(self):
+        sample_rate = 256
+        trial_length_s = 0.5
+        reshaped_trials, labels = TrialReshaper()(
+            trial_targetness_label=self.target_info,
+            timing_info=self.timing_info,
+            eeg_data=self.eeg,
+            sample_rate=sample_rate,
+            channel_map=None,
+            poststimulus_length=trial_length_s
+        )
+        trial_length_samples = int(sample_rate * trial_length_s)
+        expected_shape = (self.channel_number, len(self.target_info), trial_length_samples)
+        self.assertTrue(np.all(labels == [1, 0, 0]))
+        self.assertTrue(reshaped_trials.shape == expected_shape)
+
 
 class TestInquiryReshaper(unittest.TestCase):
 
@@ -918,6 +950,20 @@ def test_inquiry_reshaper(self):
         self.assertTrue(reshaped_data.shape == expected_shape)
         self.assertTrue(np.all(labels == self.true_labels))
 
+    def test_inquiry_reshaper_with_no_channel_map(self):
+        reshaped_data, labels, _ = InquiryReshaper()(
+            trial_targetness_label=self.target_info,
+            timing_info=self.timing_info,
+            eeg_data=self.eeg,
+            sample_rate=self.sample_rate,
+            trials_per_inquiry=self.trials_per_inquiry,
+            channel_map=None,
+            poststimulus_length=self.trial_length
+        )
+        expected_shape = (self.n_channel, self.n_inquiry, self.samples_per_inquiry)
+        self.assertTrue(reshaped_data.shape == expected_shape)
+        self.assertTrue(np.all(labels == self.true_labels))
+
     def test_inquiry_reshaper_trial_extraction(self):
         timing = [[1, 3, 4], [1, 4, 5], [1, 2, 3], [4, 5, 6]]
         # make a fake eeg data array (n_channels, n_inquiry, n_samples)