From 71a1e3520667e19f355c5e761465796a2721bc5a Mon Sep 17 00:00:00 2001
From: Craig Barnes <craig.barnes@member.fsf.org>
Date: Fri, 2 May 2014 21:36:07 +0100
Subject: [PATCH] Fix pyboard.py tests.

---
 tools/pyboard.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/tools/pyboard.py b/tools/pyboard.py
index a44710914..2f77ba37c 100644
--- a/tools/pyboard.py
+++ b/tools/pyboard.py
@@ -110,18 +110,19 @@ def run_test():
 
     pyb.exec('def apply(l, f):\r\n for item in l:\r\n  f(item)\r\n')
 
-    pyb.exec('leds=[pyb.Led(l) for l in range(1, 5)]')
+    pyb.exec('leds=[pyb.LED(l) for l in range(1, 5)]')
     pyb.exec('apply(leds, lambda l:l.off())')
 
     ## USR switch test
 
-    if True:
-        for i in range(2):
-            print("press USR button")
-            pyb.exec('while pyb.switch(): pyb.delay(10)')
-            pyb.exec('while not pyb.switch(): pyb.delay(10)')
+    pyb.exec('switch = pyb.Switch()')
+
+    for i in range(2):
+        print("press USR button")
+        pyb.exec('while switch(): pyb.delay(10)')
+        pyb.exec('while not switch(): pyb.delay(10)')
 
-        print('USR switch passed')
+    print('USR switch passed')
 
     ## accel test
 
-- 
GitLab