Skip to content

Commit d08e265

Browse files
committed
fix tests
1 parent 451325c commit d08e265

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

neo/rawio/elanrawio.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
BaseRawIO,
2727
_signal_channel_dtype,
2828
_signal_stream_dtype,
29+
_signal_buffer_dtype,
2930
_spike_channel_dtype,
3031
_event_channel_dtype,
3132
)
@@ -154,7 +155,10 @@ def _parse_header(self):
154155
n = int(round(np.log(channel_infos[0]["max_logic"] - channel_infos[0]["min_logic"]) / np.log(2)) / 8)
155156
sig_dtype = np.dtype(">i" + str(n))
156157

157-
signal_streams = np.array([("Signals", "0")], dtype=_signal_stream_dtype)
158+
# unique buffer and stream
159+
signal_buffers = np.array([("Signals", "0")], dtype=_signal_buffer_dtype)
160+
signal_streams = np.array([("Signals", "0", "0")], dtype=_signal_stream_dtype)
161+
158162

159163
sig_channels = []
160164
for c, chan_info in enumerate(channel_infos[:-2]):
@@ -166,8 +170,9 @@ def _parse_header(self):
166170
)
167171
offset = -chan_info["min_logic"] * gain + chan_info["min_physic"]
168172
stream_id = "0"
173+
buffer_id = "0"
169174
sig_channels.append(
170-
(chan_name, chan_id, self._sampling_rate, sig_dtype, chan_info["units"], gain, offset, stream_id)
175+
(chan_name, chan_id, self._sampling_rate, sig_dtype, chan_info["units"], gain, offset, stream_id, buffer_id)
171176
)
172177

173178
sig_channels = np.array(sig_channels, dtype=_signal_channel_dtype)
@@ -203,6 +208,7 @@ def _parse_header(self):
203208
self.header = {}
204209
self.header["nb_block"] = 1
205210
self.header["nb_segment"] = [1]
211+
self.header["signal_buffers"] = signal_buffers
206212
self.header["signal_streams"] = signal_streams
207213
self.header["signal_channels"] = sig_channels
208214
self.header["spike_channels"] = spike_channels

neo/rawio/mearecrawio.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def _parse_header(self):
9595
self._dtype = self.info_dict["recordings"]["dtype"]
9696

9797
signal_buffers = [("Signals", "0")] if self.load_analogsignal else []
98-
signal_streams = [("Signals", "0"), "0"] if self.load_analogsignal else []
98+
signal_streams = [("Signals", "0", "0")] if self.load_analogsignal else []
9999
signal_streams = np.array(signal_streams, dtype=_signal_stream_dtype)
100100
signal_buffers = np.array(signal_buffers, dtype=_signal_buffer_dtype)
101101

0 commit comments

Comments
 (0)