def test_video_mode_get_data_set(self, verbose=0):
with mock.patch('sys.stdout', new_callable=io.StringIO) as mock_stdout:
station = qtt.simulation.virtual_dot_array.initialize(
reinit=True, verbose=verbose)
digitizer = SimulationDigitizer(
qtt.measurements.scans.instrumentName('sdigitizer'),
model=station.model)
station.add_component(digitizer)
station.awg = SimulationAWG(
qtt.measurements.scans.instrumentName('vawg'))
station.add_component(station.awg)
sweepparams = {'gates_horz': {'P1': 1}, 'gates_vert': {'P2': 1}}
minstrument = (digitizer.name, [0])
vm = VideoMode(station,
sweepparams,
sweepranges=[120] * 2,
minstrument=minstrument,
resolution=[12] * 2,
Naverage=2)
vm.stop()
vm.updatebg()
data = vm.get_dataset()
vm.close()
self.assertIsInstance(data, list)
self.assertIsInstance(data[0], qcodes.DataSet)
self.assertEqual(data[0].measured.shape, (12, 12))
vm = VideoMode(station, ['P1', 'P2'],
sweepranges=[20] * 2,
minstrument=minstrument,
resolution=[32] * 2,
Naverage=2)
vm.stop()
vm.updatebg()
data = vm.get_dataset()
vm.close()
self.assertIsInstance(data, list)
self.assertIsInstance(data[0], qcodes.DataSet)
self.assertEqual(data[0].measured.shape, (32, 32))
for _, instrument in station.components.items():
instrument.close()
qtt.simulation.virtual_dot_array.close(verbose=verbose)
std_output = mock_stdout.getvalue()
print(std_output)
self.assertIn('VideoMode: start readout', std_output)
def test_videomode_getdataset(self):
station = qtt.simulation.virtual_dot_array.initialize(reinit=True,
verbose=0)
digitizer = SimulationDigitizer(
qtt.measurements.scans.instrumentName('sdigitizer'),
model=station.model)
station.add_component(digitizer)
station.awg = SimulationAWG(
qtt.measurements.scans.instrumentName('vawg'))
station.add_component(station.awg)
sweepparams = {'gates_horz': {'P1': 1}, 'gates_vert': {'P2': 1}}
minstrument = (digitizer.name, [0])
vm = VideoMode(station,
sweepparams,
sweepranges=[120] * 2,
minstrument=minstrument,
resolution=[12] * 2,
Naverage=2)
vm.stop()
vm.updatebg()
data = vm.get_dataset()
vm.close()
self.assertIsInstance(data, list)
self.assertIsInstance(data[0], qcodes.DataSet)
self.assertEqual(data[0].measured.shape, (12, 12))
vm = VideoMode(station, ['P1', 'P2'],
sweepranges=[20] * 2,
minstrument=minstrument,
resolution=[32] * 2,
Naverage=2)
vm.stop()
vm.updatebg()
data = vm.get_dataset()
vm.close()
self.assertIsInstance(data, list)
self.assertIsInstance(data[0], qcodes.DataSet)
self.assertEqual(data[0].measured.shape, (32, 32))
for name, instrument in station.components.items():
instrument.close()
qtt.simulation.virtual_dot_array.close(verbose=0)