test rich print

This commit is contained in:
Will McGugan 2020-04-19 14:59:10 +01:00
parent 8181845f4a
commit 6df310f46b
1 changed files with 19 additions and 0 deletions

19
tests/test_rich_print.py Normal file
View File

@ -0,0 +1,19 @@
import io
import sys
import rich
from rich.console import Console
def test_rich_print():
output = io.StringIO()
assert rich._console is None
backup_file = sys.stdout
try:
sys.stdout = output
rich.print("foo")
assert isinstance(rich._console, Console)
assert output.getvalue() == "foo\n"
finally:
sys.stdout = backup_file