diff options
author | shadchin <shadchin@yandex-team.ru> | 2022-02-10 16:44:39 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:44:39 +0300 |
commit | e9656aae26e0358d5378e5b63dcac5c8dbe0e4d0 (patch) | |
tree | 64175d5cadab313b3e7039ebaa06c5bc3295e274 /contrib/python/prompt-toolkit/py2/tests/test_shortcuts.py | |
parent | 2598ef1d0aee359b4b6d5fdd1758916d5907d04f (diff) | |
download | ydb-e9656aae26e0358d5378e5b63dcac5c8dbe0e4d0.tar.gz |
Restoring authorship annotation for <shadchin@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'contrib/python/prompt-toolkit/py2/tests/test_shortcuts.py')
-rw-r--r-- | contrib/python/prompt-toolkit/py2/tests/test_shortcuts.py | 100 |
1 files changed, 50 insertions, 50 deletions
diff --git a/contrib/python/prompt-toolkit/py2/tests/test_shortcuts.py b/contrib/python/prompt-toolkit/py2/tests/test_shortcuts.py index c677306859..8c13510ea8 100644 --- a/contrib/python/prompt-toolkit/py2/tests/test_shortcuts.py +++ b/contrib/python/prompt-toolkit/py2/tests/test_shortcuts.py @@ -1,50 +1,50 @@ -from prompt_toolkit.shortcuts import _split_multiline_prompt -from prompt_toolkit.token import Token - - -def test_split_multiline_prompt(): - # Test 1: no newlines: - tokens = [(Token, 'ab')] - has_before_tokens, before, first_input_line = _split_multiline_prompt(lambda cli: tokens) - assert has_before_tokens(None) is False - assert before(None) == [] - assert first_input_line(None) == [ - (Token, 'a'), - (Token, 'b'), - ] - - # Test 1: multiple lines. - tokens = [(Token, 'ab\ncd\nef')] - has_before_tokens, before, first_input_line = _split_multiline_prompt(lambda cli: tokens) - assert has_before_tokens(None) is True - assert before(None) == [ - (Token, 'a'), - (Token, 'b'), - (Token, '\n'), - (Token, 'c'), - (Token, 'd'), - ] - assert first_input_line(None) == [ - (Token, 'e'), - (Token, 'f'), - ] - - # Edge case 1: starting with a newline. - tokens = [(Token, '\nab')] - has_before_tokens, before, first_input_line = _split_multiline_prompt(lambda cli: tokens) - assert has_before_tokens(None) is True - assert before(None) == [] - assert first_input_line(None) == [ - (Token, 'a'), - (Token, 'b') - ] - - # Edge case 2: starting with two newlines. - tokens = [(Token, '\n\nab')] - has_before_tokens, before, first_input_line = _split_multiline_prompt(lambda cli: tokens) - assert has_before_tokens(None) is True - assert before(None) == [(Token, '\n')] - assert first_input_line(None) == [ - (Token, 'a'), - (Token, 'b') - ] +from prompt_toolkit.shortcuts import _split_multiline_prompt +from prompt_toolkit.token import Token + + +def test_split_multiline_prompt(): + # Test 1: no newlines: + tokens = [(Token, 'ab')] + has_before_tokens, before, first_input_line = _split_multiline_prompt(lambda cli: tokens) + assert has_before_tokens(None) is False + assert before(None) == [] + assert first_input_line(None) == [ + (Token, 'a'), + (Token, 'b'), + ] + + # Test 1: multiple lines. + tokens = [(Token, 'ab\ncd\nef')] + has_before_tokens, before, first_input_line = _split_multiline_prompt(lambda cli: tokens) + assert has_before_tokens(None) is True + assert before(None) == [ + (Token, 'a'), + (Token, 'b'), + (Token, '\n'), + (Token, 'c'), + (Token, 'd'), + ] + assert first_input_line(None) == [ + (Token, 'e'), + (Token, 'f'), + ] + + # Edge case 1: starting with a newline. + tokens = [(Token, '\nab')] + has_before_tokens, before, first_input_line = _split_multiline_prompt(lambda cli: tokens) + assert has_before_tokens(None) is True + assert before(None) == [] + assert first_input_line(None) == [ + (Token, 'a'), + (Token, 'b') + ] + + # Edge case 2: starting with two newlines. + tokens = [(Token, '\n\nab')] + has_before_tokens, before, first_input_line = _split_multiline_prompt(lambda cli: tokens) + assert has_before_tokens(None) is True + assert before(None) == [(Token, '\n')] + assert first_input_line(None) == [ + (Token, 'a'), + (Token, 'b') + ] |