python-progressbar/tests/test_unknown_length.py at develop · commit-0/python-progressbar · GitHub

30 lines (23 loc) · 962 Bytes

import progressbar

def test_unknown_length() -> None:

pb = progressbar.ProgressBar(

widgets=[progressbar.AnimatedMarker()],

max_value=progressbar.UnknownLength,

)

assert pb.max_value is progressbar.UnknownLength

def test_unknown_length_default_widgets() -> None:

# The default widgets picked should work without a known max_value

pb = progressbar.ProgressBar(max_value=progressbar.UnknownLength).start()

for i in range(60):

pb.update(i)

pb.finish()

def test_unknown_length_at_start() -> None:

# The default widgets should be picked after we call .start()

pb = progressbar.ProgressBar().start(max_value=progressbar.UnknownLength)

for i in range(60):

pb.update(i)

pb.finish()

pb2 = progressbar.ProgressBar().start(max_value=progressbar.UnknownLength)

for w in pb2.widgets:

print(type(w), repr(w))

assert any(isinstance(w, progressbar.Bar) for w in pb2.widgets)

Expand file tree