mirror of https://github.com/Textualize/rich.git
36 lines
919 B
Python
36 lines
919 B
Python
import io
|
|
|
|
from rich.abc import RichRenderable
|
|
from rich.console import Console
|
|
from rich.panel import Panel
|
|
from rich.text import Text
|
|
|
|
|
|
class Foo:
|
|
def __rich__(self) -> Text:
|
|
return Text("Foo")
|
|
|
|
|
|
def test_rich_cast():
|
|
foo = Foo()
|
|
console = Console(file=io.StringIO())
|
|
console.print(foo)
|
|
assert console.file.getvalue() == "Foo\n"
|
|
|
|
|
|
def test_rich_cast_container():
|
|
foo = Foo()
|
|
console = Console(file=io.StringIO(), legacy_windows=False)
|
|
console.print(Panel.fit(foo, padding=0))
|
|
assert console.file.getvalue() == "╭───╮\n│Foo│\n╰───╯\n"
|
|
|
|
|
|
def test_abc():
|
|
foo = Foo()
|
|
assert isinstance(foo, RichRenderable)
|
|
assert isinstance(Text("hello"), RichRenderable)
|
|
assert isinstance(Panel("hello"), RichRenderable)
|
|
assert not isinstance(foo, str)
|
|
assert not isinstance("foo", RichRenderable)
|
|
assert not isinstance([], RichRenderable)
|