diff --git a/artiq/test/coredevice/test_rtio.py b/artiq/test/coredevice/test_rtio.py index 8f5dcb48c..881c16ae9 100644 --- a/artiq/test/coredevice/test_rtio.py +++ b/artiq/test/coredevice/test_rtio.py @@ -12,6 +12,10 @@ from artiq.test.hardware_testbench import ExperimentCase artiq_low_latency = os.getenv("ARTIQ_LOW_LATENCY") +class PulseNotReceived(Exception): + pass + + class RTT(EnvExperiment): def build(self): self.setattr_device("core") @@ -29,7 +33,10 @@ class RTT(EnvExperiment): delay(1*us) t0 = now_mu() self.ttl_inout.pulse(1*us) - self.set_dataset("rtt", mu_to_seconds(self.ttl_inout.timestamp_mu() - t0)) + t1 = self.ttl_inout.timestamp_mu() + if t1 < 0: + raise PulseNotReceived() + self.set_dataset("rtt", mu_to_seconds(t1 - t0)) class Loopback(EnvExperiment): @@ -48,7 +55,10 @@ class Loopback(EnvExperiment): delay(1*us) t0 = now_mu() self.loop_out.pulse(1*us) - self.set_dataset("rtt", mu_to_seconds(self.loop_in.timestamp_mu() - t0)) + t1 = self.loop_in.timestamp_mu() + if t1 < 0: + raise PulseNotReceived() + self.set_dataset("rtt", mu_to_seconds(t1 - t0)) class ClockGeneratorLoopback(EnvExperiment):