def test_single_measurement_with_averaging(self, mock_waveform_thread):
# setup
settings = ExperimentSettings()
settings.averaging = 2
handler = MeasurementHandler()
return_tuple = [
(np.array([2, 3, 4, 5, 6, 7]), np.array([0, 1])),
(np.array([3, 4, 5, 6, 7, 8]), np.array([0, 1])),
]
handler.add_to_queue(self.lp.create_waveform(), settings)
with patch.object(handler, '_run_thread', side_effect=return_tuple) as method:
# perform
test_dataset = handler.single_measurement()
# assert
self.assertEqual(2, method.call_count)
np.testing.assert_array_equal(
test_dataset,
np.array([[0, 2.5, 4.5, 6.5], [1, 3.5, 5.5, 7.5]])
)
self.assertEqual(len(handler._queue), 0)
