[pypy-commit] pyrepl py3ksupport: fix up keymap creation

[RonnyPfannschmidt]

Author: Ronny Pfannschmidt <Ronny.Pfannschmidt at gmx.de>
Branch: py3ksupport
Changeset: r158:9de498f86d73
Date: 2011-11-08 23:59 +0100
http://bitbucket.org/pypy/pyrepl/changeset/9de498f86d73/

Log:	fix up keymap creation

diff --git a/pyrepl/keymap.py b/pyrepl/keymap.py
--- a/pyrepl/keymap.py
+++ b/pyrepl/keymap.py
@@ -174,7 +174,7 @@
     r = {}
     import pprint
     for key, value in keymap.items():
-        r.setdefault(key[0], {})[key[1:]] = value
+        r.setdefault(key[:1], {})[key[1:]] = value
     for key, value in r.items():
         if empty in value:
             if len(value) != 1:
diff --git a/testing/test_keymap.py b/testing/test_keymap.py
new file mode 100644
--- /dev/null
+++ b/testing/test_keymap.py
@@ -0,0 +1,10 @@
+from pyrepl.keymap import compile_keymap
+
+
+def test_compile_keymap():
+    k = compile_keymap({
+        b'a': 'test',
+        b'bc': 'test2',
+    })
+
+    assert k == {b'a': 'test', b'b': { b'c': 'test2'}}