mirror of https://github.com/Textualize/rich.git
102 lines
6.0 KiB
Python
102 lines
6.0 KiB
Python
from rich.console import Console
|
|
from rich.progress_bar import ProgressBar
|
|
from rich.segment import Segment
|
|
from rich.style import Style
|
|
|
|
from .render import render
|
|
|
|
|
|
def test_init():
|
|
bar = ProgressBar(completed=50)
|
|
repr(bar)
|
|
assert bar.percentage_completed == 50.0
|
|
|
|
|
|
def test_update():
|
|
bar = ProgressBar()
|
|
assert bar.completed == 0
|
|
assert bar.total == 100
|
|
bar.update(10, 20)
|
|
assert bar.completed == 10
|
|
assert bar.total == 20
|
|
assert bar.percentage_completed == 50
|
|
bar.update(100)
|
|
assert bar.percentage_completed == 100
|
|
|
|
|
|
expected = [
|
|
"\x1b[38;2;249;38;114m━━━━━\x1b[0m\x1b[38;2;249;38;114m╸\x1b[0m\x1b[38;5;237m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\x1b[0m",
|
|
"\x1b[38;2;249;38;114m━━━━━━\x1b[0m\x1b[38;5;237m╺\x1b[0m\x1b[38;5;237m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\x1b[0m",
|
|
]
|
|
|
|
|
|
def test_render():
|
|
bar = ProgressBar(completed=11, width=50)
|
|
bar_render = render(bar)
|
|
assert bar_render == expected[0]
|
|
bar.update(completed=12)
|
|
bar_render = render(bar)
|
|
assert bar_render == expected[1]
|
|
|
|
|
|
def test_measure():
|
|
console = Console(width=120)
|
|
bar = ProgressBar()
|
|
measurement = bar.__rich_measure__(console, console.options)
|
|
assert measurement.minimum == 4
|
|
assert measurement.maximum == 120
|
|
|
|
|
|
def test_zero_total():
|
|
# Shouldn't throw zero division error
|
|
bar = ProgressBar(total=0)
|
|
render(bar)
|
|
|
|
|
|
def test_pulse():
|
|
bar = ProgressBar(pulse=True, animation_time=10)
|
|
bar_render = render(bar)
|
|
print(repr(bar_render))
|
|
expected = "\x1b[38;2;249;38;114m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;58;58;58m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;249;38;114m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;58;58;58m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;249;38;114m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;58;58;58m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;249;38;114m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;58;58;58m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;249;38;114m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;58;58;58m━\x1b[0m\x1b[38;2;62;57;59m━\x1b[0m\x1b[38;2;76;56;63m━\x1b[0m\x1b[38;2;97;53;69m━\x1b[0m\x1b[38;2;123;51;77m━\x1b[0m\x1b[38;2;153;48;86m━\x1b[0m\x1b[38;2;183;44;94m━\x1b[0m\x1b[38;2;209;42;102m━\x1b[0m\x1b[38;2;230;39;108m━\x1b[0m\x1b[38;2;244;38;112m━\x1b[0m"
|
|
assert bar_render == expected
|
|
|
|
|
|
def test_get_pulse_segments():
|
|
bar = ProgressBar()
|
|
segments = bar._get_pulse_segments(
|
|
Style.parse("red"), Style.parse("yellow"), None, False, False
|
|
)
|
|
print(repr(segments))
|
|
expected = [
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("red")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
Segment("━", Style.parse("yellow")),
|
|
]
|
|
assert segments == expected
|
|
|
|
|
|
if __name__ == "__main__":
|
|
bar = ProgressBar(completed=11, width=50)
|
|
bar_render = render(bar)
|
|
print(repr(bar_render))
|
|
bar.update(completed=12)
|
|
bar_render = render(bar)
|
|
print(repr(bar_render))
|