fix tests and clean

Author: samuelgarcia

neo/rawio/cedrawio.py

@@ -57,19 +57,17 @@ def _parse_header(self):
 
         self.smrx_file = sonpy.lib.SonFile(sName=str(self.filename), bReadOnly=True)
         smrx = self.smrx_file
-        
+
         self._time_base = smrx.GetTimeBase()
 
         channel_infos = []
         signal_channels = []
         spike_channels = []
         self._all_spike_ticks = {}
-        
+
         for chan_ind in range(smrx.MaxChannels()):
             chan_type = smrx.ChannelType(chan_ind)
             chan_id = str(chan_ind)
-            #~ print(chan_type)
-            #~ continue
             if chan_type == sonpy.lib.DataType.Adc:
                 physical_chan = smrx.PhysicalChannel(chan_ind)
                 divide = smrx.ChannelDivide(chan_ind)
@@ -86,7 +84,7 @@ def _parse_header(self):
                 offset = smrx.GetChannelOffset(chan_ind)
                 units = smrx.GetChannelUnits(chan_ind)
                 ch_name = smrx.GetChannelTitle(chan_ind)
-                
+
                 dtype = 'int16'
                 # set later after grouping
                 stream_id = '0'
@@ -101,12 +99,12 @@ def _parse_header(self):
                 divide = smrx.ChannelDivide(chan_ind)
                 # here we don't use filter (sonpy.lib.MarkerFilter()) so we get all marker
                 wave_marks = smrx.ReadWaveMarks(chan_ind, int(max_time/divide), 0, max_time)
-                
-                # here we load in memory all spike once for all because the access is really slow
+
+                # here we load in memory all spike once because the access is really slow
                 # with the ReadWaveMarks
                 spike_ticks = np.array([t.Tick for t in wave_marks])
                 spike_codes = np.array([t.Code1 for t in wave_marks])
-                
+
                 unit_ids = np.unique(spike_codes)
                 for unit_id in unit_ids:
                     name = f'{ch_name}#{unit_id}'
@@ -115,8 +113,7 @@ def _parse_header(self):
                     mask = spike_codes == unit_id
                     self._all_spike_ticks[spike_chan_id] = spike_ticks[mask]
 
-        signal_channels = np.array(signal_channels, dtype=_signal_channel_dtype)
-        
+        signal_channels = np.array(signal_channels, dtype=_signal_channel_dtype)       
 
         # channels are grouped into stream if they have a common start, stop, size, divide and sampling_rate
         channel_infos = np.array(channel_infos,
@@ -145,11 +142,9 @@ def _parse_header(self):
         self._seg_t_stop = -np.inf
         for info in self.stream_info:
             self._seg_t_start = min(self._seg_t_start,
-                                    #~ info['first_time'] / info['sampling_rate'])
                                     info['first_time'] * self._time_base)
-                                    
+
             self._seg_t_stop = max(self._seg_t_stop,
-                                   #~ info['max_time'] / info['sampling_rate'])
                                    info['max_time'] * self._time_base)
 
         self.header = {}
@@ -174,7 +169,6 @@ def _get_signal_size(self, block_index, seg_index, stream_index):
 
     def _get_signal_t_start(self, block_index, seg_index, stream_index):
         info = self.stream_info[stream_index]
-        #~ t_start = info['first_time'] / info['sampling_rate']
         t_start = info['first_time'] * self._time_base
         return t_start
 
@@ -209,12 +203,11 @@ def _get_analogsignal_chunk(self, block_index, seg_index, i_start, i_stop,
             sigs[:, i] = sig
 
         return sigs
-    
+   
     def _spike_count(self, block_index, seg_index, unit_index):
         unit_id = self.header['spike_channels'][unit_index]['id']
         spike_ticks = self._all_spike_ticks[unit_id]
         return spike_ticks.size
-        
 
     def _get_spike_timestamps(self, block_index, seg_index, unit_index, t_start, t_stop):
         unit_id = self.header['spike_channels'][unit_index]['id']
@@ -232,3 +225,6 @@ def _rescale_spike_timestamp(self, spike_timestamps, dtype):
         spike_times *= self._time_base
         return spike_times
 
+    def _get_spike_raw_waveforms(self, block_index, seg_index,
+                                 spike_channel_index, t_start, t_stop):
+        return None