Commit missing parts of 501de30.

m-labs · Mar 18, 2016 · 891e3cf · 891e3cf
1 parent 501de30
commit 891e3cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/artiq/test/coredevice/test_rtio.py b/artiq/test/coredevice/test_rtio.py
@@ -7,6 +7,7 @@
 
 from artiq.experiment import *
 from artiq.test.hardware_testbench import ExperimentCase
+from artiq.coredevice import exceptions
 
 
 artiq_low_latency = os.getenv("ARTIQ_LOW_LATENCY")
@@ -285,7 +286,7 @@ def test_address_collision(self):
 
     def test_watchdog(self):
         # watchdog only works on the device
-        with self.assertRaises(IOError):
+        with self.assertRaises(exceptions.WatchdogExpired):
             self.execute(Watchdog)
 
     @unittest.skipUnless(artiq_low_latency,