1 files changed, 99 insertions, 0 deletions
diff --git a/tests/test_bar.py b/tests/test_bar.py
new file mode 100644
index 0000000..46f8e4e
--- /dev/null
+++ b/tests/test_bar.py
@@ -0,0 +1,99 @@
+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():
+    bar = ProgressBar()
+    measurement = bar.__rich_measure__(None, 120)
+    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"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("red"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+        Segment("━", Style.parse("yellow"), False),
+    ]
+    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))