def test_overwrite_multiple_progress_bars_with_section_outputs(ansi_io):
output1 = ansi_io.section()
output2 = ansi_io.section()
bar1 = ProgressBar(output1, 50, 0)
bar2 = ProgressBar(output2, 50, 0)
bar1.start()
bar2.start()
bar2.advance()
bar1.advance()
output = [
" 0/50 [>---------------------------] 0%",
" 0/50 [>---------------------------] 0%",
"\x1b[1A\x1b[0J 1/50 [>---------------------------] 2%",
"\x1b[2A\x1b[0J 1/50 [>---------------------------] 2%",
"\x1b[1A\x1b[0J 1/50 [>---------------------------] 2%",
" 1/50 [>---------------------------] 2%",
]
expected = "\n".join(output) + "\n"
assert expected == ansi_io.fetch_error()
def test_regress_below_min(ansi_io):
bar = ProgressBar(ansi_io, 10, 0)
bar.set_progress(1)
bar.advance(-1)
bar.advance(-1)
output = [
" 1/10 [==>-------------------------] 10%",
" 0/10 [>---------------------------] 0%",
]
expected = "\x0D" + "\x0D".join(output)
assert expected == ansi_io.fetch_error()
def test_advance_over_max(ansi_io):
bar = ProgressBar(ansi_io, 10)
bar.set_progress(9)
bar.advance()
bar.advance()
output = [
" 9/10 [=========================>--] 90%",
" 10/10 [============================] 100%",
" 11/11 [============================] 100%",
]
expected = "\x0D" + "\x0D".join(output)
assert expected == ansi_io.fetch_error()
def test_customizations(ansi_io):
bar = ProgressBar(ansi_io, 10, 0)
bar.set_bar_width(10)
bar.set_bar_character("_")
bar.set_empty_bar_character(" ")
bar.set_progress_character("/")
bar.set_format(" %current%/%max% [%bar%] %percent:3s%%")
bar.start()
bar.advance()
output = [" 0/10 [/ ] 0%", " 1/10 [_/ ] 10%"]
expected = "\x0D" + "\x0D".join(output)
assert expected == ansi_io.fetch_error()
def test_overwrite_with_section_output(ansi_io):
bar = ProgressBar(ansi_io.section(), 50, 0)
bar.start()
bar.display()
bar.advance()
bar.advance()
output = [
" 0/50 [>---------------------------] 0%",
" 0/50 [>---------------------------] 0%",
" 1/50 [>---------------------------] 2%",
" 2/50 [=>--------------------------] 4%",
]
expected = "\n\x1b[1A\x1b[0J".join(output) + "\n"
assert expected == ansi_io.fetch_error()
def test_percent_not_hundred_before_complete(ansi_io):
bar = ProgressBar(ansi_io, 200, 0)
bar.start()
bar.display()
bar.advance(199)
bar.advance()
output = [
" 0/200 [>---------------------------] 0%",
" 0/200 [>---------------------------] 0%",
" 199/200 [===========================>] 99%",
" 200/200 [============================] 100%",
]
expected = "\x0D" + "\x0D".join(output)
assert expected == ansi_io.fetch_error()
def test_percent(ansi_io):
bar = ProgressBar(ansi_io, 50, 0)
bar.start()
bar.display()
bar.advance()
bar.advance()
output = [
" 0/50 [>---------------------------] 0%",
" 0/50 [>---------------------------] 0%",
" 1/50 [>---------------------------] 2%",
" 2/50 [=>--------------------------] 4%",
]
expected = "\x0D" + "\x0D".join(output)
assert expected == ansi_io.fetch_error()
def test_multiline_format(ansi_io):
bar = ProgressBar(ansi_io, 3, 0)
bar.set_format("%bar%\nfoobar")
bar.start()
bar.advance()
bar.clear()
bar.finish()
output = [
"\033[1A>---------------------------\nfoobar",
"\033[1A=========>------------------\nfoobar ",
"\033[1A \n ",
"\033[1A============================\nfoobar ",
]
expected = "\x0D" + "\x0D".join(output)
assert expected == ansi_io.fetch_error()
def test_set_current_progress(ansi_io):
bar = ProgressBar(ansi_io, 50, 0)
bar.start()
bar.display()
bar.advance()
bar.set_progress(15)
bar.set_progress(25)
output = [
" 0/50 [>---------------------------] 0%",
" 0/50 [>---------------------------] 0%",
" 1/50 [>---------------------------] 2%",
" 15/50 [========>-------------------] 30%",
" 25/50 [==============>-------------] 50%",
]
expected = "\x0D" + "\x0D".join(output)
assert expected == ansi_io.fetch_error()
def test_overwrite_with_shorter_line(ansi_io):
bar = ProgressBar(ansi_io, 50, 0)
bar.set_format(" %current%/%max% [%bar%] %percent:3s%%")
bar.start()
bar.display()
bar.advance()
# Set shorter format
bar.set_format(" %current%/%max% [%bar%]")
bar.advance()
output = [
" 0/50 [>---------------------------] 0%",
" 0/50 [>---------------------------] 0%",
" 1/50 [>---------------------------] 2%",
" 2/50 [=>--------------------------] ",
]
expected = "\x0D" + "\x0D".join(output)
assert expected == ansi_io.fetch_error()
def test_format(ansi_io):
output = [
" 0/10 [>---------------------------] 0%",
" 10/10 [============================] 100%",
" 10/10 [============================] 100%",
]
expected = "\x0D" + "\x0D".join(output)
# max in construct, no format
ansi_io.clear_error()
bar = ProgressBar(ansi_io, 10)
bar.start()
bar.advance(10)
bar.finish()
assert expected == ansi_io.fetch_error()
# max in start, no format
ansi_io.clear_error()
bar = ProgressBar(ansi_io)
bar.start(10)
bar.advance(10)
bar.finish()
assert expected == ansi_io.fetch_error()
# max in construct, explicit format before
ansi_io.clear_error()
bar = ProgressBar(ansi_io, 10)
bar.set_format("normal")
bar.start()
bar.advance(10)
bar.finish()
assert expected == ansi_io.fetch_error()
# max in start, explicit format before
ansi_io.clear_error()
bar = ProgressBar(ansi_io)
bar.set_format("normal")
bar.start(10)
bar.advance(10)
bar.finish()
assert expected == ansi_io.fetch_error()
def test_non_decorated_output(io):
bar = ProgressBar(io, 200, 0)
bar.start()
for i in range(200):
bar.advance()
bar.finish()
expected = "\n".join([
" 0/200 [>---------------------------] 0%",
" 20/200 [==>-------------------------] 10%",
" 40/200 [=====>----------------------] 20%",
" 60/200 [========>-------------------] 30%",
" 80/200 [===========>----------------] 40%",
" 100/200 [==============>-------------] 50%",
" 120/200 [================>-----------] 60%",
" 140/200 [===================>--------] 70%",
" 160/200 [======================>-----] 80%",
" 180/200 [=========================>--] 90%",
" 200/200 [============================] 100%",
])
assert expected == io.fetch_error()
