def test_measure_segment_no_data_raises_warning():
expected_data = np.array([])
waveform = {'bla': 1, 'ble': 2, 'blu': 3}
number_of_averages = 100
read_channels = [0, 1]
with patch('qtt.instrument_drivers.simulation_instruments.SimulationDigitizer', \
spec=SimulationDigitizer) as simulation_digitizer, patch('warnings.warn') as warn_mock:
simulation_digitizer.measuresegment.return_value = expected_data
actual_data = measuresegment(waveform, number_of_averages, simulation_digitizer, read_channels)
warn_mock.assert_called_once_with('measuresegment: received empty data array')
np.testing.assert_array_equal(expected_data, actual_data)
def test_measure_segment_simulator_has_correct_output():
expected_data = np.array([1, 2, 3, 4])
waveform = {'bla': 1, 'ble': 2, 'blu': 3}
number_of_averages = 100
read_channels = [0, 1]
with patch('qtt.instrument_drivers.simulation_instruments.SimulationDigitizer',
spec=SimulationDigitizer) as simulation_digitizer:
simulation_digitizer.measuresegment.return_value = expected_data
actual_data = measuresegment(waveform, number_of_averages, simulation_digitizer, read_channels)
np.testing.assert_array_equal(actual_data, expected_data)
simulation_digitizer.measuresegment.assert_called_with(waveform, channels=read_channels)
def test_measure_segment_uhfli_has_correct_output():
expected_data = np.array([1, 2, 3, 4])
waveform = {'bla': 1, 'ble': 2, 'blu': 3}
number_of_averages = 100
read_channels = [0, 1]
with patch.object(zhinst.utils, 'create_api_session', return_value=3 * (MagicMock(),)), \
patch('qtt.measurements.scans.measure_segment_uhfli') as measure_segment_mock:
uhfli_digitizer = ZIUHFLI('test', 'dev1234')
measure_segment_mock.return_value = expected_data
actual_data = measuresegment(waveform, number_of_averages, uhfli_digitizer, read_channels)
np.testing.assert_array_equal(actual_data, expected_data)
measure_segment_mock.assert_called_with(uhfli_digitizer, waveform, read_channels, number_of_averages)
uhfli_digitizer.close()
def test_measure_segment_m4i_has_correct_output():
expected_data = np.array([1, 2, 3, 4])
waveform = {'bla': 1, 'ble': 2, 'blu': 3}
number_of_averages = 100
read_channels = [0, 1]
with patch('qtt.measurements.scans.measuresegment_m4i') as measure_segment_mock:
m4i_digitizer = M4i('test')
measure_segment_mock.return_value = expected_data
actual_data = measuresegment(waveform, number_of_averages, m4i_digitizer, read_channels)
np.testing.assert_array_equal(actual_data, expected_data)
measure_segment_mock.assert_called_with(m4i_digitizer, waveform, read_channels,
2000, number_of_averages, process=True)
m4i_digitizer.close()
