X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/7c0d801cb4b9231b71fb9c7fc69af58f43c1595f..55f9f0cbc8dcf45f8d484b2f931c5c254d4404d1:/wxPython/tests/gridtest.py

diff --git a/wxPython/tests/gridtest.py b/wxPython/tests/gridtest.py
index b6549d3757..8635051d5c 100644
--- a/wxPython/tests/gridtest.py
+++ b/wxPython/tests/gridtest.py
@@ -83,9 +83,8 @@ class MyCellEditor(wxPyGridCellEditor):
         """
         key = evt.GetKeyCode()
         print "KeyCode:", key
-        # For linux the first range means number keys in main keyboard, and
-        # the second range means numeric keypad keys, with num-lock on.
-        if key in range(48,58) or key in range(326,336):
+        if (key in range(ord('0'),ord('9')+1) or
+            key in range(WXK_NUMPAD0, WXK_NUMPAD9+1)):
             return true
         else:
             return false
@@ -99,14 +98,13 @@ class MyCellEditor(wxPyGridCellEditor):
         key = evt.GetKeyCode()
         print "StartingKey", key
         ch = None
-        if key in [WXK_NUMPAD0, WXK_NUMPAD1, WXK_NUMPAD2, WXK_NUMPAD3, WXK_NUMPAD4,
-                   WXK_NUMPAD5, WXK_NUMPAD6, WXK_NUMPAD7, WXK_NUMPAD8, WXK_NUMPAD9]:
+        if key in range(WXK_NUMPAD0, WXK_NUMPAD9+1):
             ch = ch = chr(ord('0') + key - WXK_NUMPAD0)
 
         elif key < 256 and key >= 0 and chr(key) in string.printable:
             ch = chr(key)
             if not evt.ShiftDown():
-                ch = string.lower(ch)
+                ch = ch.lower()
 
         if ch is not None:
             # Replace the text.  Other option would be to append it.