def test_evaluate_variable_missing(self) -> None: e = ExpressionScalar('a * b + c') params = { 'b': 1.5 } with self.assertRaises(ExpressionVariableMissingException): e.evaluate_numeric(**params)
def test_evaluate_numeric(self) -> None: e = ExpressionScalar('a * b + c') params = {'a': 2, 'b': 1.5, 'c': -7} self.assertEqual(2 * 1.5 - 7, e.evaluate_numeric(**params)) with self.assertRaises(NonNumericEvaluation): params['a'] = sympify('h') e.evaluate_numeric(**params)
def test_indexing(self): e = ExpressionScalar('a[i] * c') params = {'a': np.array([1, 2, 3]), 'i': 1, 'c': 2} self.assertEqual(e.evaluate_numeric(**params), 2 * 2) params['a'] = [1, 2, 3] self.assertEqual(e.evaluate_numeric(**params), 2 * 2) params['a'] = np.array([[1, 2, 3], [4, 5, 6]]) np.testing.assert_equal(e.evaluate_numeric(**params), 2 * np.array([4, 5, 6]))
def test_evaluate_numpy(self): e = ExpressionScalar('a * b + c') params = { 'a': 2 * np.ones(4), 'b': 1.5 * np.ones(4), 'c': -7 * np.ones(4) } np.testing.assert_equal((2 * 1.5 - 7) * np.ones(4), e.evaluate_numeric(**params)) e = ExpressionScalar('a * b + c') params = {'a': np.array(2), 'b': np.array(1.5), 'c': np.array(-7)} np.testing.assert_equal((2 * 1.5 - 7), e.evaluate_numeric(**params))
def from_expression( cls, expression: ExpressionScalar, duration: float, channel: ChannelID) -> Union['FunctionWaveform', ConstantWaveform]: if expression.variables: return cls(expression, duration, channel) else: return ConstantWaveform(amplitude=expression.evaluate_numeric(), duration=duration, channel=channel)
class AtomicMultiChannelPulseTemplate(AtomicPulseTemplate, ParameterConstrainer): """Combines multiple PulseTemplates that are defined on different channels into an AtomicPulseTemplate.""" def __init__(self, *subtemplates: Union[AtomicPulseTemplate, MappingTuple, MappingPulseTemplate], external_parameters: Optional[Set[str]] = None, identifier: Optional[str] = None, parameter_constraints: Optional[List] = None, measurements: Optional[List[MeasurementDeclaration]] = None, registry: PulseRegistryType = None, duration: Union[str, Expression, bool] = False) -> None: """Parallels multiple AtomicPulseTemplates of the same duration. The duration equality check is performed on construction by default. If the duration keyword argument is given the check is performed on instantiation (when build_waveform is called). duration can be a Expression to enforce a certain duration or True for an unspecified duration. Args: *subtemplates: Positional arguments are subtemplates to combine. identifier: Forwarded to AtomicPulseTemplate.__init__ parameter_constraints: Forwarded to ParameterConstrainer.__init__ measurements: Forwarded to AtomicPulseTemplate.__init__ duration: Enforced duration of the pulse template on instantiation. build_waveform checks all sub-waveforms have this duration. If True the equality of durations is only checked durtin instantiation not construction. external_parameters: No functionality. (Deprecated) """ AtomicPulseTemplate.__init__(self, identifier=identifier, measurements=measurements) ParameterConstrainer.__init__( self, parameter_constraints=parameter_constraints) self._subtemplates = [ st if isinstance(st, PulseTemplate) else MappingPulseTemplate.from_tuple(st) for st in subtemplates ] for subtemplate in self._subtemplates: if isinstance(subtemplate, AtomicPulseTemplate): continue elif isinstance(subtemplate, MappingPulseTemplate): if isinstance(subtemplate.template, AtomicPulseTemplate): continue else: raise TypeError( 'Non atomic subtemplate of MappingPulseTemplate: {}'. format(subtemplate.template)) else: raise TypeError( 'Non atomic subtemplate: {}'.format(subtemplate)) if not self._subtemplates: raise ValueError('Cannot create empty MultiChannelPulseTemplate') defined_channels = [st.defined_channels for st in self._subtemplates] # check there are no intersections between channels for i, channels_i in enumerate(defined_channels): for j, channels_j in enumerate(defined_channels[i + 1:]): if channels_i & channels_j: raise ChannelMappingException( 'subtemplate {}'.format(i + 1), 'subtemplate {}'.format(i + 2 + j), (channels_i | channels_j).pop()) if external_parameters is not None: warnings.warn( "external_parameters is an obsolete argument and will be removed in the future.", category=DeprecationWarning) if not duration: duration = self._subtemplates[0].duration for subtemplate in self._subtemplates[1:]: if almost_equal(duration.sympified_expression, subtemplate.duration.sympified_expression): continue else: raise ValueError( 'Could not assert duration equality of {} and {}'. format(duration, subtemplate.duration)) self._duration = None elif duration is True: self._duration = None else: self._duration = ExpressionScalar(duration) self._register(registry=registry) @property def duration(self) -> ExpressionScalar: if self._duration: return self._duration else: return self._subtemplates[0].duration @property def parameter_names(self) -> Set[str]: return set.union(self.measurement_parameters, self.constrained_parameters, *(st.parameter_names for st in self._subtemplates), self._duration.variables if self._duration else ()) @property def subtemplates( self ) -> Sequence[Union[AtomicPulseTemplate, MappingPulseTemplate]]: return self._subtemplates @property def defined_channels(self) -> Set[ChannelID]: return set.union(*(st.defined_channels for st in self._subtemplates)) @property def measurement_names(self) -> Set[str]: return super().measurement_names.union(*(st.measurement_names for st in self._subtemplates)) def build_waveform( self, parameters: Dict[str, numbers.Real], channel_mapping: Dict[ChannelID, Optional[ChannelID]] ) -> Optional[Waveform]: self.validate_parameter_constraints(parameters=parameters) sub_waveforms = [] for subtemplate in self.subtemplates: sub_waveform = subtemplate.build_waveform( parameters, channel_mapping=channel_mapping) if sub_waveform is not None: sub_waveforms.append(sub_waveform) if len(sub_waveforms) == 0: return None if len(sub_waveforms) == 1: waveform = sub_waveforms[0] else: waveform = MultiChannelWaveform(sub_waveforms) if self._duration: expected_duration = self._duration.evaluate_numeric(**parameters) if not isclose(expected_duration, waveform.duration): raise ValueError( 'The duration does not ' 'equal the expected duration', expected_duration, waveform.duration) return waveform def get_measurement_windows( self, parameters: Dict[str, numbers.Real], measurement_mapping: Dict[str, Optional[str]]) -> List[MeasurementWindow]: measurements = super().get_measurement_windows( parameters=parameters, measurement_mapping=measurement_mapping) for st in self.subtemplates: measurements.extend( st.get_measurement_windows( parameters=parameters, measurement_mapping=measurement_mapping)) return measurements def requires_stop(self, parameters: Dict[str, Parameter], conditions: Dict[str, 'Condition']) -> bool: return any( st.requires_stop(parameters, conditions) for st in self._subtemplates) def get_serialization_data(self, serializer: Optional[Serializer] = None ) -> Dict[str, Any]: data = super().get_serialization_data(serializer) data['subtemplates'] = self.subtemplates if serializer: # compatibility to old serialization routines, deprecated data = dict() data['subtemplates'] = [ serializer.dictify(subtemplate) for subtemplate in self.subtemplates ] if self.parameter_constraints: data['parameter_constraints'] = [ str(constraint) for constraint in self.parameter_constraints ] if self.measurement_declarations: data['measurements'] = self.measurement_declarations return data @classmethod def deserialize(cls, serializer: Optional[Serializer] = None, **kwargs) -> 'AtomicMultiChannelPulseTemplate': subtemplates = kwargs['subtemplates'] del kwargs['subtemplates'] if serializer: # compatibility to old serialization routines, deprecated subtemplates = [serializer.deserialize(st) for st in subtemplates] return cls(*subtemplates, **kwargs) @property def integral(self) -> Dict[ChannelID, ExpressionScalar]: expressions = dict() for subtemplate in self._subtemplates: expressions.update(subtemplate.integral) return expressions