blob: 1fc8ae4398157a8cb3723d5e9523e0d3e171cc85 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
from __future__ import annotations
from prompt_toolkit.formatted_text import fragment_list_to_text
from prompt_toolkit.layout import to_window
from prompt_toolkit.widgets import Button
def _to_text(button: Button) -> str:
control = to_window(button).content
return fragment_list_to_text(control.text())
def test_defaulf_button():
button = Button("Exit")
assert _to_text(button) == "< Exit >"
def test_custom_button():
button = Button("Exit", left_symbol="[", right_symbol="]")
assert _to_text(button) == "[ Exit ]"
|
