From 1cb8f642b4d87d889b4e95de5b44eb545ec89d95 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Wed, 7 Sep 2016 17:37:49 +0800
Subject: [PATCH] test: level-based TTL APIs (#218)

---
 artiq/test/coredevice/test_rtio.py | 65 ++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/artiq/test/coredevice/test_rtio.py b/artiq/test/coredevice/test_rtio.py
index 8a261d5dc..f60abd643 100644
--- a/artiq/test/coredevice/test_rtio.py
+++ b/artiq/test/coredevice/test_rtio.py
@@ -172,6 +172,65 @@ class LoopbackCount(EnvExperiment):
         self.set_dataset("count", self.loop_in.count())
 
 
+class IncorrectLevel(Exception):
+    pass
+
+
+class Level(EnvExperiment):
+    def build(self):
+        self.setattr_device("core")
+        self.setattr_device("loop_in")
+        self.setattr_device("loop_out")
+
+    @kernel
+    def run(self):
+        self.core.reset()
+        self.loop_in.input()
+        self.loop_out.output()
+        delay(5*us)
+
+        self.loop_out.off()
+        delay(5*us)
+        if self.loop_in.sample_get_nonrt():
+            raise IncorrectLevel
+
+        self.loop_out.on()
+        delay(5*us)
+        if not self.loop_in.sample_get_nonrt():
+            raise IncorrectLevel
+
+
+class Watch(EnvExperiment):
+    def build(self):
+        self.setattr_device("core")
+        self.setattr_device("loop_in")
+        self.setattr_device("loop_out")
+
+    @kernel
+    def run(self):
+        self.core.reset()
+        self.loop_in.input()
+        self.loop_out.output()
+        delay(5*us)
+
+        self.loop_out.off()
+        delay(5*us)
+        if not self.loop_in.watch_stay_off():
+            raise IncorrectLevel
+        delay(10*us)
+        if not self.loop_in.watch_done():
+            raise IncorrectLevel
+
+        delay(10*us)
+        if not self.loop_in.watch_stay_off():
+            raise IncorrectLevel
+        delay(3*us)
+        self.loop_out.on()
+        delay(10*us)
+        if self.loop_in.watch_done():
+            raise IncorrectLevel
+
+
 class Underflow(EnvExperiment):
     def build(self):
         self.setattr_device("core")
@@ -309,6 +368,12 @@ class CoredeviceTest(ExperimentCase):
         count = self.dataset_mgr.get("count")
         self.assertEqual(count, npulses)
 
+    def test_level(self):
+        self.execute(Level)
+
+    def test_watch(self):
+        self.execute(Watch)
+
     def test_underflow(self):
         with self.assertRaises(RTIOUnderflow):
             self.execute(Underflow)