From fa1a9bc9fd1683f1fb68739efc1aef303c8a1d2d Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Mon, 25 Aug 2014 18:44:10 +0100
Subject: [PATCH] tests: Add test for pyb.disable_irq and pyb.enable_irq.

---
 tests/pyb/irq.py     | 22 ++++++++++++++++++++++
 tests/pyb/irq.py.exp |  2 ++
 2 files changed, 24 insertions(+)
 create mode 100644 tests/pyb/irq.py
 create mode 100644 tests/pyb/irq.py.exp

diff --git a/tests/pyb/irq.py b/tests/pyb/irq.py
new file mode 100644
index 000000000..42d276568
--- /dev/null
+++ b/tests/pyb/irq.py
@@ -0,0 +1,22 @@
+import pyb
+
+def test_irq():
+    # test basic disable/enable
+    i1 = pyb.disable_irq()
+    print(i1)
+    pyb.enable_irq() # by default should enable IRQ
+
+    # check that interrupts are enabled by waiting for ticks
+    pyb.delay(10)
+
+    # check nested disable/enable
+    i1 = pyb.disable_irq()
+    i2 = pyb.disable_irq()
+    print(i1, i2)
+    pyb.enable_irq(i2)
+    pyb.enable_irq(i1)
+
+    # check that interrupts are enabled by waiting for ticks
+    pyb.delay(10)
+
+test_irq()
diff --git a/tests/pyb/irq.py.exp b/tests/pyb/irq.py.exp
new file mode 100644
index 000000000..aea065f04
--- /dev/null
+++ b/tests/pyb/irq.py.exp
@@ -0,0 +1,2 @@
+True
+True False