summaryrefslogtreecommitdiffstats
path: root/tests/test_widgets.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_widgets.py')
-rw-r--r--tests/test_widgets.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/tests/test_widgets.py b/tests/test_widgets.py
new file mode 100644
index 0000000..bf89a25
--- /dev/null
+++ b/tests/test_widgets.py
@@ -0,0 +1,18 @@
+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 ]"