Skip to content

Commit a04f580

Browse files
committed
add test for imagesequence
1 parent d919e65 commit a04f580

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

neo/test/iotest/test_neomatlabio.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
from neo.core.analogsignal import AnalogSignal
1010
from neo.core.irregularlysampledsignal import IrregularlySampledSignal
11-
from neo import Block, Segment, SpikeTrain
11+
from neo import Block, Segment, SpikeTrain, ImageSequence
1212
from neo.test.iotest.common_io_test import BaseTestIO
1313
from neo.io.neomatlabio import NeoMatlabIO
1414

@@ -32,10 +32,17 @@ def test_write_read_single_spike(self):
3232
spiketrain1.annotate(yep='yop')
3333
sig1 = AnalogSignal([4, 5, 6] * pq.A, sampling_period=1 * pq.ms)
3434
irrsig1 = IrregularlySampledSignal([0, 1, 2] * pq.ms, [4, 5, 6] * pq.A)
35+
img_sequence_array = [[[column for column in range(2)] for _ in range(2)]
36+
for _ in range(2)]
37+
image_sequence = ImageSequence(img_sequence_array, units='dimensionless',
38+
sampling_rate=1 * pq.Hz,
39+
spatial_scale=1 * pq.micrometer)
3540
block1.segments.append(seg)
3641
seg.spiketrains.append(spiketrain1)
3742
seg.analogsignals.append(sig1)
3843
seg.irregularlysampledsignals.append(irrsig1)
44+
seg.imagesequences.append(image_sequence)
45+
3946

4047
# write block
4148
filename = self.get_local_path('matlabiotestfile.mat')
@@ -56,6 +63,9 @@ def test_write_read_single_spike(self):
5663
block2.segments[0].irregularlysampledsignals[0].magnitude)
5764
assert_array_equal(block1.segments[0].irregularlysampledsignals[0].times,
5865
block2.segments[0].irregularlysampledsignals[0].times)
66+
67+
assert_array_equal(block1.segments[0].imagesequences[0],
68+
block2.segments[0].imagesequences[0])
5969

6070
# test annotations
6171
spiketrain2 = block2.segments[0].spiketrains[0]

0 commit comments

Comments
 (0)