Browse Source

Fixed annotation test to inlcude only those annotations currently used.

Michael Denker 11 months ago
parent
commit
af8f2ecf07
1 changed files with 8 additions and 9 deletions
  1. 8 9
      code/reachgraspio/test_reachgraspio.py

+ 8 - 9
code/reachgraspio/test_reachgraspio.py

@@ -77,22 +77,21 @@ class RGIOTestCase(unittest.TestCase):
                         self.assertTrue(all(np.diff(valid_ids) >= 0))
 
     def test_rejection_annotations_present(self):
-        for block in self.blocks:
+        for block in [self.blocks[1]]:#self.blocks:
             for seg in block.segments:
                 for anasig in seg.analogsignals:
                     if anasig.annotations['neural_signal']:
-                        for key in ['file_origin', 'connector_ID', 'connector_pinID',
-                                    'nsx_hi_freq_order', 'nsx_lo_freq_order', 'nsx_hi_freq_type',
-                                    'nsx_lo_freq_type', 'description', 'nsx',
+                        for key in ['channel_ids', 'channel_names',
+                                    'hi_pass_freq', 'lo_pass_freq',
+                                    'hi_pass_order', 'lo_pass_order', 'filter_type',
                                     'electrode_reject_IFC', 'electrode_reject_LFC',
-                                    'electrode_reject_HFC']:
+                                    'electrode_reject_HFC', 'coordinates_x', 'coordinates_y']:
                             self.assertIn(key, anasig.array_annotations)
 
                 for st in seg.spiketrains:
-                    for key in ['electrode_reject_IFC', 'electrode_reject_LFC',
-                                'electrode_reject_HFC','nev_dig_factor', 'nb_sorted_units',
-                                'nev_hi_freq_order', 'nev_hi_freq_type', 'nev_lo_freq_order',
-                                'nev_lo_freq_type']:
+                    for key in ['id', 'channel_id', 'unit_id', 'unit_tag',
+                                'electrode_reject_IFC', 'electrode_reject_LFC', 'electrode_reject_HFC',
+                                'sua', 'mua', 'noise', 'coordinate_x', 'coordinate_y', 'connector_aligned_id']:
                         self.assertIn(key, st.annotations)
 
     def test_event_annotations(self):