summaryrefslogtreecommitdiffstats
path: root/tests/test_screen.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_screen.py')
-rw-r--r--tests/test_screen.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/test_screen.py b/tests/test_screen.py
new file mode 100644
index 0000000..7596c3a
--- /dev/null
+++ b/tests/test_screen.py
@@ -0,0 +1,12 @@
+from rich.console import Console
+from rich.screen import Screen
+
+
+def test_screen():
+ console = Console(color_system=None, width=20, height=5)
+ with console.capture() as capture:
+ console.print(Screen("foo\nbar\nbaz\nfoo\nbar\nbaz\foo"))
+ result = capture.get()
+ print(repr(result))
+ expected = "foo \nbar \nbaz \nfoo \nbar "
+ assert result == expected