workaround for ImportError when curses is unavailable

zariiii9003 · zariiii9003 · commit 1fc6ae2c8a0a · 2021-04-13T20:38:42.000+02:00
diff --git a/test/test_viewer.py b/test/test_viewer.py
@@ -36,7 +36,7 @@
 import pytest
 
 import can
-from can.viewer import KEY_ESC, KEY_SPACE, CanViewer, parse_args
+from can.viewer import CanViewer, parse_args
 
 
 # Allow the curses module to be missing (e.g. on PyPy on Windows)
@@ -81,16 +81,18 @@ def nodelay(_bool):
         pass
 
     def getch(self):
+        assert curses is not None
+
         self.key_counter += 1
         if self.key_counter == 1:
             # Send invalid key
             return -1
         elif self.key_counter == 2:
             return ord("c")  # Clear
         elif self.key_counter == 3:
-            return KEY_SPACE  # Pause
+            return curses.ascii.SP  # Pause
         elif self.key_counter == 4:
-            return KEY_SPACE  # Unpause
+            return curses.ascii.SP  # Unpause
         elif self.key_counter == 5:
             return ord("s")  # Sort
 
@@ -101,7 +103,7 @@ def getch(self):
         elif self.key_counter <= 200:
             return curses.KEY_UP
 
-        return KEY_ESC
+        return curses.ascii.ESC
 
 
 @unittest.skipUnless(CURSES_AVAILABLE, "curses might be missing on some platforms")
