summaryrefslogtreecommitdiffstats
path: root/tests/render.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/render.py')
-rw-r--r--tests/render.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/tests/render.py b/tests/render.py
new file mode 100644
index 0000000..a2435c5
--- /dev/null
+++ b/tests/render.py
@@ -0,0 +1,24 @@
+import io
+import re
+
+from rich.console import Console, RenderableType
+
+
+re_link_ids = re.compile(r"id=[\d\.\-]*?;.*?\x1b")
+
+
+def replace_link_ids(render: str) -> str:
+ """Link IDs have a random ID and system path which is a problem for
+ reproducible tests.
+
+ """
+ return re_link_ids.sub("id=0;foo\x1b", render)
+
+
+def render(renderable: RenderableType, no_wrap: bool = False) -> str:
+ console = Console(
+ width=100, file=io.StringIO(), color_system="truecolor", legacy_windows=False
+ )
+ console.print(renderable, no_wrap=no_wrap)
+ output = replace_link_ids(console.file.getvalue())
+ return output