def test_padding_less_than_sched_duration(self):
"""Test that the until arg is respected even for less than the input schedule duration."""
delay = 10
sched = (Delay(delay, DriveChannel(0)) +
Delay(delay, DriveChannel(0)).shift(20))
ref_sched = (sched | pulse.Delay(5, DriveChannel(0)).shift(10))
self.assertEqual(pad(sched, until=15), ref_sched)
def test_padding_until_less(self):
"""Test padding until time that is less than schedule duration."""
delay = pulse.Delay(10)
sched = (delay(DriveChannel(0)).shift(10) + delay(DriveChannel(1)))
ref_sched = (sched | delay(DriveChannel(0))
| pulse.Delay(5)(DriveChannel(1)).shift(10))
self.assertEqual(pad(sched, until=15), ref_sched)
def test_padding_until_less(self):
"""Test padding until time that is less than schedule duration."""
delay = 10
sched = (Delay(delay, DriveChannel(0)).shift(10) +
Delay(delay, DriveChannel(1)))
ref_sched = (sched | Delay(delay, DriveChannel(0))
| Delay(5, DriveChannel(1)).shift(10))
self.assertEqual(pad(sched, until=15), ref_sched)
def test_padding_until_greater(self):
"""Test padding until time that is greater than schedule duration."""
delay = pulse.Delay(10)
sched = (delay(DriveChannel(0)).shift(10) + delay(DriveChannel(1)))
ref_sched = (sched | delay(DriveChannel(0))
| pulse.Delay(30)(DriveChannel(0)).shift(20)
| pulse.Delay(40)(DriveChannel(1)).shift(10))
self.assertEqual(pad(sched, until=50), ref_sched)
def test_padding_supplied_channels(self):
"""Test padding of only specified channels."""
delay = 10
sched = (Delay(delay, DriveChannel(0)).shift(10) +
Delay(delay, DriveChannel(1)))
ref_sched = (sched | Delay(delay, DriveChannel(0))
| Delay(2 * delay, DriveChannel(2)))
channels = [DriveChannel(0), DriveChannel(2)]
self.assertEqual(pad(sched, channels=channels), ref_sched)
def test_padding_until_greater(self):
"""Test padding until time that is greater than schedule duration."""
delay = 10
sched = (Delay(delay, DriveChannel(0)).shift(10) +
Delay(delay, DriveChannel(1)))
ref_sched = (sched | Delay(delay, DriveChannel(0))
| Delay(30, DriveChannel(0)).shift(20)
| Delay(40, DriveChannel(1)).shift(10))
self.assertEqual(pad(sched, until=50), ref_sched)
def test_padding_schedule(self):
"""Test padding schedule."""
delay = 10
sched = (Delay(delay, DriveChannel(0)).shift(10) +
Delay(delay, DriveChannel(0)).shift(10) +
Delay(delay, DriveChannel(1)).shift(10))
ref_sched = (sched | Delay(delay, DriveChannel(0))
| Delay(delay, DriveChannel(0)).shift(20)
| Delay(delay, DriveChannel(1))
| Delay(2 * delay, DriveChannel(1)).shift(20))
self.assertEqual(pad(sched), ref_sched)
def test_padding_supplied_channels(self):
"""Test padding of only specified channels."""
delay = pulse.Delay(10)
double_delay = pulse.Delay(20)
sched = (delay(DriveChannel(0)).shift(10) + delay(DriveChannel(1)))
ref_sched = (sched | delay(DriveChannel(0))
| double_delay(DriveChannel(2)))
channels = [DriveChannel(0), DriveChannel(2)]
self.assertEqual(pad(sched, channels=channels), ref_sched)
def test_padding_schedule(self):
"""Test padding schedule."""
delay = pulse.Delay(10)
double_delay = pulse.Delay(20)
sched = (delay(pulse.DriveChannel(0)).shift(10) +
delay(pulse.DriveChannel(0)).shift(10) +
delay(pulse.DriveChannel(1)).shift(10))
ref_sched = (sched | delay(pulse.DriveChannel(0))
| delay(pulse.DriveChannel(0)).shift(20)
| delay(pulse.DriveChannel(1))
| double_delay(pulse.DriveChannel(1)).shift(20))
self.assertEqual(pad(sched), ref_sched)
def test_padding_schedule_inverse_order(self):
"""Test padding schedule is insensitive to order in which commands were added.
This test is the same as `test_adding_schedule` but the order by channel
in which commands were added to the schedule to be padded has been reversed.
"""
delay = 10
sched = (Delay(delay, DriveChannel(1)).shift(10) +
Delay(delay, DriveChannel(0)).shift(10) +
Delay(delay, DriveChannel(0)).shift(10))
ref_sched = (sched | Delay(delay, DriveChannel(0))
| Delay(delay, DriveChannel(0)).shift(20)
| Delay(delay, DriveChannel(1))
| Delay(2 * delay, DriveChannel(1)).shift(20))
self.assertEqual(pad(sched), ref_sched)
def test_padding_empty_schedule(self):
"""Test padding of empty schedule."""
self.assertEqual(pulse.Schedule(), pad(pulse.Schedule()))
