mirror of https://github.com/Textualize/rich.git
75 lines
1.8 KiB
Python
75 lines
1.8 KiB
Python
import pytest
|
|
|
|
from rich.console import Console
|
|
from rich.measure import Measurement
|
|
from rich.rule import Rule
|
|
from rich.spinner import Spinner
|
|
from rich.text import Text
|
|
|
|
|
|
def test_spinner_create():
|
|
Spinner("dots")
|
|
with pytest.raises(KeyError):
|
|
Spinner("foobar")
|
|
|
|
|
|
def test_spinner_render():
|
|
time = 0.0
|
|
|
|
def get_time():
|
|
nonlocal time
|
|
return time
|
|
|
|
console = Console(
|
|
width=80, color_system=None, force_terminal=True, get_time=get_time
|
|
)
|
|
console.begin_capture()
|
|
spinner = Spinner("dots", "Foo")
|
|
console.print(spinner)
|
|
time += 80 / 1000
|
|
console.print(spinner)
|
|
result = console.end_capture()
|
|
print(repr(result))
|
|
expected = "⠋ Foo\n⠙ Foo\n"
|
|
assert result == expected
|
|
|
|
|
|
def test_spinner_update():
|
|
time = 0.0
|
|
|
|
def get_time():
|
|
nonlocal time
|
|
return time
|
|
|
|
console = Console(width=20, force_terminal=True, get_time=get_time)
|
|
console.begin_capture()
|
|
spinner = Spinner("dots")
|
|
console.print(spinner)
|
|
|
|
spinner.update(text="Bar", style="green", speed=2)
|
|
time += 80 / 1000
|
|
console.print(spinner)
|
|
|
|
spinner.update(text=Rule("Bar"))
|
|
time += 80 / 1000
|
|
console.print(spinner)
|
|
|
|
result = console.end_capture()
|
|
print(repr(result))
|
|
expected = f"⠋\n\x1b[32m⠙\x1b[0m Bar\n\x1b[32m⠸\x1b[0m \x1b[92m────── \x1b[0mBar\x1b[92m ───────\x1b[0m\n"
|
|
assert result == expected
|
|
|
|
|
|
def test_rich_measure():
|
|
console = Console(width=80, color_system=None, force_terminal=True)
|
|
spinner = Spinner("dots", "Foo")
|
|
min_width, max_width = Measurement.get(console, console.options, spinner)
|
|
assert min_width == 3
|
|
assert max_width == 5
|
|
|
|
|
|
def test_spinner_markup():
|
|
spinner = Spinner("dots", "[bold]spinning[/bold]")
|
|
assert isinstance(spinner.text, Text)
|
|
assert str(spinner.text) == "spinning"
|