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)])