def test_from_binary_segment(self):
ch_a = np.asarray(100 + np.arange(32), dtype=np.uint16)
ch_b = np.asarray(1000 + np.arange(32), dtype=np.uint16)
marker_a = np.ones(16, dtype=bool)
marker_b = np.asarray(list(range(5)) + list(range(6)) + list(range(5)),
dtype=np.uint16)
segment = TaborSegment(ch_a=ch_a,
ch_b=ch_b,
marker_a=marker_a,
marker_b=marker_b)
binary = segment.get_as_binary()
reconstructed = TaborSegment.from_binary_segment(binary)
self.assertEqual(segment, reconstructed)
def test_read_waveforms(self):
self.channel_pair._amend_segments(self.segments)
waveforms = self.channel_pair.read_waveforms()
segments = [
TaborSegment.from_binary_segment(waveform)
for waveform in waveforms
]
expected = [self.zero_segment, *self.segments]
for ex, r in zip(expected, segments):
ex1, ex2 = ex.data_a, ex.data_b
r1, r2 = r.data_a, r.data_b
np.testing.assert_equal(ex1, r1)
np.testing.assert_equal(ex2, r2)
self.assertEqual(expected, segments)
