def test_equispaced_with_short_duration(self):
"""Test equispaced context with duration shorter than the schedule duration."""
d0 = pulse.DriveChannel(0)
sched = pulse.Schedule()
for _ in range(3):
sched.append(Delay(10, d0), inplace=True)
sched = transforms.align_equispaced(sched, duration=20)
reference = pulse.Schedule()
reference.insert(0, Delay(10, d0), inplace=True)
reference.insert(10, Delay(10, d0), inplace=True)
reference.insert(20, Delay(10, d0), inplace=True)
self.assertEqual(sched, reference)
def test_equispaced_with_multiple_channels_short_duration(self):
"""Test equispaced context with multiple channels and duration shorter than the total
duration."""
d0 = pulse.DriveChannel(0)
d1 = pulse.DriveChannel(1)
sched = pulse.Schedule()
sched.append(Delay(10, d0), inplace=True)
sched.append(Delay(20, d1), inplace=True)
sched = transforms.align_equispaced(sched, duration=20)
reference = pulse.Schedule()
reference.insert(0, Delay(10, d0), inplace=True)
reference.insert(0, Delay(20, d1), inplace=True)
self.assertEqual(sched, reference)