mirror of
https://github.com/Textualize/rich.git
synced 2025-08-04 01:58:24 +00:00
37 lines
845 B
Python
37 lines
845 B
Python
import io
|
|
|
|
import rich
|
|
from rich.console import Console
|
|
|
|
|
|
def test_get_console():
|
|
console = rich.get_console()
|
|
assert isinstance(console, Console)
|
|
|
|
|
|
def test_rich_print():
|
|
console = rich.get_console()
|
|
output = io.StringIO()
|
|
backup_file = console.file
|
|
try:
|
|
console.file = output
|
|
rich.print("foo")
|
|
rich.print("foo\n")
|
|
rich.print("foo\n\n")
|
|
assert output.getvalue() == "foo\nfoo\n\nfoo\n\n\n"
|
|
finally:
|
|
console.file = backup_file
|
|
|
|
|
|
def test_rich_print_X():
|
|
console = rich.get_console()
|
|
output = io.StringIO()
|
|
backup_file = console.file
|
|
try:
|
|
console.file = output
|
|
rich.print("foo")
|
|
rich.print("fooX")
|
|
rich.print("fooXX")
|
|
assert output.getvalue() == "foo\nfooX\nfooXX\n"
|
|
finally:
|
|
console.file = backup_file
|