Merge pull request #1213 from alejoe91/oe-legacy-ts-fix

Add `ignore_timestamps_errors` argument to OpenEphysRawIO

Author: JuliaSprenger

neo/rawio/openephysrawio.py

@@ -68,9 +68,10 @@ class OpenEphysRawIO(BaseRawIO):
     extensions = ['continuous', 'openephys', 'spikes', 'events', 'xml']
     rawmode = 'one-dir'
 
-    def __init__(self, dirname=''):
+    def __init__(self, dirname='', ignore_timestamps_errors=False):
         BaseRawIO.__init__(self)
         self.dirname = dirname
+        self._ignore_timestamps_errors = ignore_timestamps_errors
 
     def _source_name(self):
         return self.dirname
@@ -115,9 +116,11 @@ def _parse_header(self):
 
                 # check for continuity (no gaps)
                 diff = np.diff(data_chan['timestamp'])
-                assert np.all(diff == RECORD_SIZE), \
-                    'Not continuous timestamps for {}. ' \
-                    'Maybe because recording was paused/stopped.'.format(continuous_filename)
+                if not np.all(diff == RECORD_SIZE) and not self._ignore_timestamps_errors:
+                    raise ValueError(
+                        'Not continuous timestamps for {}. ' \
+                        'Maybe because recording was paused/stopped.'.format(continuous_filename)
+                    )
 
                 if seg_index == 0:
                     # add in channel list

neo/test/rawiotest/test_openephysrawio.py

@@ -20,8 +20,14 @@ def test_raise_error_if_discontinuous_files(self):
         # It must raise an error
         reader = OpenEphysRawIO(dirname=self.get_local_path(
             'openephys/OpenEphys_SampleData_2_(multiple_starts)'))
-        with self.assertRaises(Exception):
+        with self.assertRaises(ValueError):
             reader.parse_header()
+        # if ignore_timestamps_errors=True, no exception is raised
+        reader = OpenEphysRawIO(dirname=self.get_local_path(
+            'openephys/OpenEphys_SampleData_2_(multiple_starts)'),
+                                ignore_timestamps_errors=True)
+        reader.parse_header()
+
 
     def test_raise_error_if_strange_timestamps(self):
         # In this dataset CH32 have strange timestamps