Skip to content

Commit 6084740

Browse files
committed
[nixio] adjust test to match expected proxy loading behaviour
1 parent c82c301 commit 6084740

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

neo/test/iotest/test_nixio.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,8 +168,10 @@ def compare_signal_dalist(self, neosig, nixdalist):
168168
def compare_eests_mtags(self, eestlist, mtaglist):
169169
self.assertEqual(len(eestlist), len(mtaglist))
170170
for eest in eestlist:
171-
if isinstance(eest, (EventProxy, EpochProxy, SpikeTrainProxy)):
171+
if isinstance(eest, (EventProxy, EpochProxy)):
172172
eest = eest.load()
173+
elif isinstance(eest, SpikeTrainProxy):
174+
eest = eest.load(load_waveforms=True)
173175
mtag = mtaglist[eest.annotations["nix_name"]]
174176
if isinstance(eest, Epoch):
175177
self.compare_epoch_mtag(eest, mtag)

0 commit comments

Comments
 (0)