Skip to content

Commit f99e6e3

Browse files
committed
don't use assert
1 parent 0d9e0c7 commit f99e6e3

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

neo/rawio/examplerawio.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -287,17 +287,21 @@ def _get_analogsignal_chunk(self, block_index, seg_index, i_start, i_stop,
287287
if i_stop is None:
288288
i_stop = 100000
289289

290-
assert i_start >= 0, "I don't like your jokes"
291-
assert i_stop <= 100000, "I don't like your jokes"
290+
if i_start < 0 or i_stop > 100000:
291+
# some check
292+
raise IndexError("I don't like your jokes")
293+
292294
if channel_indexes is None:
293295
nb_chan = 8
294296
elif isinstance(channel_indexes, slice):
295297
channel_indexes = np.arange(8, dtype='int')[channel_indexes]
296298
nb_chan = len(channel_indexes)
297299
else:
298300
channel_indexes = np.asarray(channel_indexes)
299-
assert np.all(channel_indexes >= 0), 'bad boy'
300-
assert np.all(channel_indexes < 8), 'big bad wolf'
301+
if any(channel_indexes < 0):
302+
raise IndexError('bad boy')
303+
if any(channel_indexes >= 8):
304+
raise IndexError('big bad wolf')
301305
nb_chan = len(channel_indexes)
302306

303307
raw_signals = np.zeros((i_stop - i_start, nb_chan), dtype='int16')

0 commit comments

Comments
 (0)