Use get_key()

skrub-wreckers · Jan 24, 2016 · 3365a09 · 3365a09
1 parent 8953bbb
commit 3365a09
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 16 deletions.
diff --git a/tests/gate_test.py b/tests/gate_test.py
@@ -15,10 +15,7 @@
             print led_on, reading.val
             time.sleep(0.1)
 
-            try:
-                c = w.keys.get_nowait()
-                if c == ' ':
-                    led_on = not led_on
-                    led.write(not led_on)
-            except:
-                pass
+            c = w.get_key()
+            if c == ' ':
+                led_on = not led_on
+                led.write(not led_on)
diff --git a/tests/odometer_vis.py b/tests/odometer_vis.py
@@ -37,12 +37,8 @@ def to_cv(val):
                 # print time.time(), odo.val
                 logwriter.writerow((time.time(),) + odo.val)
 
-                try:
-                    c = w.keys.get_nowait()
-                except Empty:
-                    pass
-                else:
-                    if c == 'r':
-                        odo.override_position(0, 0, 0)
-                    elif c == 'q':
-                        break
+                c = w.get_key()
+                if c == 'r':
+                    odo.override_position(12, 12, np.pi / 8)
+                elif c == 'q':
+                    break