def setUp(self) -> None:
self.maxDiff = None
self.if_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'bar'})
self.else_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'hugo'})
self.template = BranchPulseTemplate('foo_condition', self.if_dummy, self.else_dummy)
self.sequencer = DummySequencer()
self.block = DummyInstructionBlock()
def test_is_interruptable(self) -> None:
if_dummy = DummyPulseTemplate(num_channels=3, is_interruptable=True)
else_dummy = DummyPulseTemplate(num_channels=3)
template = BranchPulseTemplate('foo_condition', if_dummy, else_dummy)
self.assertFalse(template.is_interruptable)
if_dummy = DummyPulseTemplate(num_channels=3, is_interruptable=True)
else_dummy = DummyPulseTemplate(num_channels=3, is_interruptable=True)
template = BranchPulseTemplate('foo_condition', if_dummy, else_dummy)
self.assertTrue(template.is_interruptable)
class BranchPulseTemplateSerializationTests(unittest.TestCase):
def setUp(self) -> None:
self.maxDiff = None
self.if_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'bar'})
self.else_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'hugo'})
self.template = BranchPulseTemplate('foo_condition', self.if_dummy, self.else_dummy)
def test_get_serialization_data(self) -> None:
expected_data = dict(
if_branch_template=str(id(self.if_dummy)),
else_branch_template=str(id(self.else_dummy)),
condition='foo_condition'
)
serializer = DummySerializer()
serialized_data = self.template.get_serialization_data(serializer)
self.assertEqual(expected_data, serialized_data)
def test_deserialize(self) -> None:
base_data = dict(
if_branch_template=str(id(self.if_dummy)),
else_branch_template=str(id(self.else_dummy)),
condition='foo_condition',
identifier='hugo'
)
serializer = DummySerializer()
serializer.subelements[str(id(self.if_dummy))] = self.if_dummy
serializer.subelements[str(id(self.else_dummy))] = self.else_dummy
template = BranchPulseTemplate.deserialize(serializer, **base_data)
self.assertEqual('hugo', template.identifier)
serialized_data = template.get_serialization_data(serializer)
del base_data['identifier']
self.assertEqual(base_data, serialized_data)
def test_deserialize(self) -> None:
base_data = dict(
if_branch_template=str(id(self.if_dummy)),
else_branch_template=str(id(self.else_dummy)),
condition='foo_condition',
identifier='hugo'
)
serializer = DummySerializer()
serializer.subelements[str(id(self.if_dummy))] = self.if_dummy
serializer.subelements[str(id(self.else_dummy))] = self.else_dummy
template = BranchPulseTemplate.deserialize(serializer, **base_data)
self.assertEqual('hugo', template.identifier)
serialized_data = template.get_serialization_data(serializer)
del base_data['identifier']
self.assertEqual(base_data, serialized_data)
class BranchPulseTemplateSequencingTests(unittest.TestCase):
def setUp(self) -> None:
self.maxDiff = None
self.if_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'bar'})
self.else_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'hugo'})
self.template = BranchPulseTemplate('foo_condition', self.if_dummy, self.else_dummy)
self.sequencer = DummySequencer()
self.block = DummyInstructionBlock()
def test_requires_stop_parameters_dont_conditions_dont(self) -> None:
conditions = dict(foo_condition=DummyCondition(requires_stop=False))
parameters = dict(foo=DummyParameter(326.272),
bar=DummyParameter(-2624.23),
hugo=DummyParameter(3.532))
self.assertFalse(self.template.requires_stop(parameters, conditions))
def test_requires_stop_parameters_do_conditions_dont(self) -> None:
conditions = dict(foo_condition=DummyCondition(requires_stop=False))
parameters = dict(foo=DummyParameter(326.272),
bar=DummyParameter(-2624.23, requires_stop=True),
hugo=DummyParameter(3.532))
self.assertFalse(self.template.requires_stop(parameters, conditions))
def test_requires_stop_parameters_dont_conditions_do(self) -> None:
conditions = dict(foo_condition=DummyCondition(requires_stop=True))
parameters = dict(foo=DummyParameter(326.272),
bar=DummyParameter(-2624.23),
hugo=DummyParameter(3.532))
self.assertTrue(self.template.requires_stop(parameters, conditions))
def test_requires_stop_parameters_do_conditions_do(self) -> None:
conditions = dict(foo_condition=DummyCondition(requires_stop=True))
parameters = dict(foo=DummyParameter(326.272, requires_stop=True),
bar=DummyParameter(-2624.23),
hugo=DummyParameter(3.532))
self.assertTrue(self.template.requires_stop(parameters, conditions))
def test_requires_stop_condition_missing(self) -> None:
conditions = dict(bar_condition=DummyCondition(requires_stop=True))
parameters = dict(foo=DummyParameter(326.272),
bar=DummyParameter(-2624.23),
hugo=DummyParameter(3.532))
with self.assertRaises(ConditionMissingException):
self.template.requires_stop(parameters, conditions)
def test_requires_stop_parameters_missing(self) -> None:
conditions = dict(foo_condition=DummyCondition(requires_stop=False))
parameters = dict(foo=DummyParameter(326.272),
hugo=DummyParameter(3.532))
self.assertFalse(self.template.requires_stop(parameters, conditions))
def test_build_sequence(self) -> None:
foo_condition = DummyCondition()
conditions = dict(foo_condition=foo_condition)
parameters = dict(foo=DummyParameter(326.272),
bar=DummyParameter(-2624.23),
hugo=DummyParameter(3.532))
self.template.build_sequence(self.sequencer, parameters, conditions, self.block)
self.assertFalse(foo_condition.loop_call_data)
self.assertEqual(
dict(
delegator=self.template,
if_branch=self.if_dummy,
else_branch=self.else_dummy,
sequencer=self.sequencer,
parameters=parameters,
conditions=conditions,
instruction_block=self.block
),
foo_condition.branch_call_data
)
self.assertFalse(self.sequencer.sequencing_stacks)
self.assertFalse(self.block.instructions)
def test_build_sequence_condition_missing(self) -> None:
conditions = dict(bar_condition=DummyCondition(requires_stop=True))
parameters = dict(foo=DummyParameter(326.272),
bar=DummyParameter(-2624.23),
hugo=DummyParameter(3.532))
with self.assertRaises(ConditionMissingException):
self.template.build_sequence(self.sequencer, parameters, conditions, self.block)
def test_build_sequence_parameter_missing(self) -> None:
foo_condition = DummyCondition()
conditions = dict(foo_condition=foo_condition)
parameters = dict(foo=DummyParameter(326.272),
bar=DummyParameter(-2624.23))
self.template.build_sequence(self.sequencer, parameters, conditions, self.block)
self.assertFalse(foo_condition.loop_call_data)
self.assertEqual(
dict(
delegator=self.template,
if_branch=self.if_dummy,
else_branch=self.else_dummy,
sequencer=self.sequencer,
parameters=parameters,
conditions=conditions,
instruction_block=self.block
),
foo_condition.branch_call_data
)
self.assertFalse(self.sequencer.sequencing_stacks)
self.assertFalse(self.block.instructions)
def test_str(self) -> None:
if_dummy = DummyPulseTemplate(num_channels=3, is_interruptable=True)
else_dummy = DummyPulseTemplate(num_channels=3)
template = BranchPulseTemplate('foo_condition', if_dummy, else_dummy)
self.assertIsInstance(str(template), str)
def test_num_channels(self) -> None:
if_dummy = DummyPulseTemplate(num_channels=3)
else_dummy = DummyPulseTemplate(num_channels=3)
template = BranchPulseTemplate('foo_condition', if_dummy, else_dummy)
self.assertEqual(3, template.num_channels)
def test_parameter_names_and_declarations(self) -> None:
if_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'bar'})
else_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'hugo'})
template = BranchPulseTemplate('foo_condition', if_dummy, else_dummy)
self.assertEqual({'foo', 'bar', 'hugo'}, template.parameter_names)
self.assertEqual({ParameterDeclaration(name) for name in {'foo', 'bar', 'hugo'}}, template.parameter_declarations)
def test_identifier(self) -> None:
if_dummy = DummyPulseTemplate(num_channels=3)
else_dummy = DummyPulseTemplate(num_channels=3)
template = BranchPulseTemplate('foo_condition', if_dummy, else_dummy, identifier='hugo')
self.assertEqual('hugo', template.identifier)
def setUp(self) -> None:
self.maxDiff = None
self.if_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'bar'})
self.else_dummy = DummyPulseTemplate(num_channels=3, parameter_names={'foo', 'hugo'})
self.template = BranchPulseTemplate('foo_condition', self.if_dummy, self.else_dummy)
def test_wrong_num_channel_composition(self) -> None:
if_dummy = DummyPulseTemplate(num_channels=2)
else_dummy = DummyPulseTemplate(num_channels=5)
with self.assertRaises(ValueError):
BranchPulseTemplate('foo_condition', if_dummy, else_dummy)