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