def test_measurement_windows(self) -> None:
pulse = FunctionPulseTemplate(5, 5, measurements=[('mw', 0, 5)])
windows = pulse.get_measurement_windows(
parameters={}, measurement_mapping={'mw': 'asd'})
self.assertEqual([('asd', 0, 5)], windows)
self.assertEqual(pulse.measurement_declarations, [('mw', 0, 5)])
def test_measurement_windows_with_parameters(self) -> None:
pulse = FunctionPulseTemplate(5, 'length', measurements=[('mw', 1, '(1+length)/2')])
parameters = dict(length=100)
windows = pulse.get_measurement_windows(parameters, measurement_mapping={'mw': 'asd'})
self.assertEqual(windows, [('asd', 1, 101/2)])
declared = pulse.measurement_declarations
self.assertEqual(declared, [('mw', 1, '(1+length)/2')])
def test_multiple_measurement_windows(self) -> None:
pulse = FunctionPulseTemplate(5,
'length',
measurements=[('A', 0, '(1+length)/2'),
('A', 1, 3),
('B', 'begin', 2)])
parameters = dict(length=5, begin=1)
measurement_mapping = dict(A='A', B='C')
windows = pulse.get_measurement_windows(
parameters=parameters, measurement_mapping=measurement_mapping)
expected = [('A', 0, 3), ('A', 1, 3), ('C', 1, 2)]
self.assertEqual(sorted(windows), sorted(expected))
self.assertEqual(pulse.measurement_declarations,
[('A', 0, '(1+length)/2'), ('A', 1, 3),
('B', 'begin', 2)])
def test_no_measurement_windows(self) -> None:
pulse = FunctionPulseTemplate(5, 5)
windows = pulse.get_measurement_windows({}, {'mw': 'asd'})
self.assertEqual([], windows)
self.assertEqual([], pulse.measurement_declarations)
