diff options
Diffstat (limited to 'extra_tests/test_pyrepl/test_reader.py')
| -rw-r--r-- | extra_tests/test_pyrepl/test_reader.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/extra_tests/test_pyrepl/test_reader.py b/extra_tests/test_pyrepl/test_reader.py new file mode 100644 index 0000000000..4b93ffa598 --- /dev/null +++ b/extra_tests/test_pyrepl/test_reader.py @@ -0,0 +1,9 @@ + +def test_process_prompt(): + from pyrepl.reader import Reader + r = Reader(None) + assert r.process_prompt("hi!") == ("hi!", 3) + assert r.process_prompt("h\x01i\x02!") == ("hi!", 2) + assert r.process_prompt("hi\033[11m!") == ("hi\033[11m!", 3) + assert r.process_prompt("h\x01i\033[11m!\x02") == ("hi\033[11m!", 1) + assert r.process_prompt("h\033[11m\x01i\x02!") == ("h\033[11mi!", 2) |