Add __pow__ convenience repetition operation.

qutech · Aug 25, 2023 · 935dd30 · 935dd30
1 parent e36e4cc
commit 935dd30
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/qupulse/pulses/pulse_template.py b/qupulse/pulses/pulse_template.py
@@ -96,6 +96,10 @@ def __rmatmul__(self, other: MappingTuple) -> 'SequencePulseTemplate':
 
         return SequencePulseTemplate.concatenate(other, self)
 
+    def __pow__(self, power: ExpressionLike):
+        """This is a convenience wrapper for :func:`.with_repetition`."""
+        return self.with_repetition(power)
+
     @property
     @abstractmethod
     def integral(self) -> Dict[ChannelID, ExpressionScalar]:

diff --git a/tests/pulses/pulse_template_tests.py b/tests/pulses/pulse_template_tests.py
@@ -376,6 +376,11 @@ def test_matmul(self):
             self.assertEqual(a @ b, 'concat')
             mock_concatenate.assert_called_once_with(a, b)
 
+    def test_pow(self):
+        pt = PulseTemplateStub()
+        pow_pt = pt ** 5
+        self.assertEqual(pow_pt, pt.with_repetition(5))
+
     def test_rmatmul(self):
         a = PulseTemplateStub()
         b = (1, 2, 3)