Add ignore_timestamps_errors flag

Author: alejoe91

neo/rawio/openephysrawio.py

@@ -10,7 +10,6 @@
 
 import os
 import re
-import warnings
 
 import numpy as np
 
@@ -69,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
@@ -116,8 +116,8 @@ def _parse_header(self):
 
                 # check for continuity (no gaps)
                 diff = np.diff(data_chan['timestamp'])
-                if not np.all(diff == RECORD_SIZE):
-                    warnings.warn(
+                if not np.all(diff == RECORD_SIZE) and not self._ignore_timestamps_errors:
+                    raise Exception(
                         'Not continuous timestamps for {}. ' \
                         'Maybe because recording was paused/stopped.'.format(continuous_filename)
                     )

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.assertWarns(UserWarning):
+        with self.assertRaises(Exception):
             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