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)