class FunctionPulseTest(unittest.TestCase):
def setUp(self):
self.s = 'a + b'
self.s2 = 'c'
self.pars = dict(a=1, b=2, c=3)
self.fpt = FunctionPulseTemplate(self.s, self.s2)
def test_get_pulse_length(self):
self.assertEqual(self.fpt.get_pulse_length(self.pars), 3)
def test_get_measurement_windows(self):
self.assertEqual(self.fpt.get_measurement_windows(self.pars), None)
fpt2 = FunctionPulseTemplate(self.s, self.s2, measurement=True)
self.assertEqual(fpt2.get_measurement_windows(self.pars), [(0, 3)])
def test_serialization_data(self):
expected_data = dict(type='FunctionPulseTemplate',
parameter_names=set(['a', 'b', 'c']),
duration_expression=self.s2,
expression=self.s,
measurement=False)
self.assertEqual(expected_data, self.fpt.get_serialization_data(DummySerializer()))
def setUp(self):
self.s = 'a + b'
self.s2 = 'c'
self.pars = dict(a=1, b=2, c=3)
self.fpt = FunctionPulseTemplate(self.s, self.s2)
def test_get_measurement_windows(self):
self.assertEqual(self.fpt.get_measurement_windows(self.pars), None)
fpt2 = FunctionPulseTemplate(self.s, self.s2, measurement=True)
self.assertEqual(fpt2.get_measurement_windows(self.pars), [(0, 3)])
