def test_insert_process(self):
b = StreamBuffer(1.0, [100, 10], 1000.0)
b.suppress_mode = 'off'
frame = usb_packet_factory(0, 1)
self.assertEqual(0, b.status()['device_sample_id']['value'])
b.insert(frame)
self.assertEqual(126, b.status()['device_sample_id']['value'])
self.assertEqual(0, b.status()['sample_id']['value'])
b.process()
self.assertEqual((0, 126), b.sample_id_range)
self.assertEqual((0.0, 1.0), b.limits_time)
self.assertEqual((-874, 126), b.limits_samples)
self.assertEqual(126, b.time_to_sample_id(1.0))
self.assertEqual(1.0, b.sample_id_to_time(126))
self.assertEqual(0.999, b.sample_id_to_time(125))
self.assertEqual(b.limits_time[0],
b.sample_id_to_time(b.limits_samples[0]))
self.assertEqual(b.limits_time[1],
b.sample_id_to_time(b.limits_samples[1]))
self.assertEqual(126, b.status()['sample_id']['value'])
samples = b.samples_get(0, 126, ['raw'])
raw_data = b.samples_get(0, 126, 'raw')
np.testing.assert_allclose(samples['signals']['raw']['value'],
raw_data)
expect = np.arange(126 * 2, dtype=np.uint16).reshape((126, 2))
np.testing.assert_allclose(expect, np.right_shift(raw_data, 2))
np.testing.assert_allclose(expect, b.data_buffer[0:126 * 2].reshape(
(126, 2)))
data = b.data_get(0, 126)
np.testing.assert_allclose(expect[:, 0], data[:, 0]['mean'])
def test_init(self):
b = StreamBuffer(1.0, [10, 10], 1000.0)
self.assertIsNotNone(b)
self.assertEqual(1000, len(b))
self.assertEqual(1000.0, b.output_sampling_frequency)
self.assertEqual((0, 0), b.sample_id_range)
self.assertEqual((0.0, 1.0), b.limits_time)
self.assertEqual((-1000, 0), b.limits_samples)
self.assertEqual(0, b.time_to_sample_id(1.0))
self.assertEqual(1.0, b.sample_id_to_time(0))
del b
