Merge pull request #77 from lysnikolaou/pyrepl-fix-disp-str-ctrl

Fix disp_str for control characters

Author: pablogsal

Lib/_pyrepl/reader.py

@@ -62,9 +62,12 @@ def disp_str(buffer: str) -> tuple[str, list[int]]:
     for c in buffer:
         if unicodedata.category(c).startswith("C"):
             c = r"\u%04x" % ord(c)
+            b.append(1)
+            b.extend([0] * (len(c) - 1))
+        else:
+            b.append(1)
+            b.extend([0] * (str_width(c) - 1))
         s.append(c)
-        b.append(1)
-        b.extend([0] * (str_width(c) - 1))
     return "".join(s), b
 
 

Lib/test/test_pyrepl.py

@@ -371,6 +371,12 @@ def test_history_search(self):
         output = multiline_input(reader)
         self.assertEqual(output, "1+1")
 
+    def test_control_character(self):
+        events = code_to_events("c\x1d\n")
+        reader = self.prepare_reader(events)
+        output = multiline_input(reader)
+        self.assertEqual(output, "c\x1d")
+
 
 class TestPyReplCompleter(TestCase):
     def prepare_reader(self, events, namespace):