def test_callback_does_nothing_with_no_message(
self, mock_manager: mock.MagicMock, mocker: MockFixture) -> None:
mock_print = mocker.patch("builtins.print")
progress_bar = ui.ProgressBar("test")
progress_bar.callback(somevalue="2")
mock_print.assert_not_called()
def test_callback_outputs_messages(self, mock_manager: mock.MagicMock,
mocker: MockFixture) -> None:
mock_print = mocker.patch("builtins.print")
progress_bar = ui.ProgressBar("test")
progress_bar.callback(message="a message")
mock_print.assert_called_with("test - info: a message")
def test_updates_counter(self, mock_manager: mock.MagicMock) -> None:
progress_bar = ui.ProgressBar("test")
mock_update_progress(progress_bar, 500)
mock_update_progress(progress_bar, 1, 0)
assert progress_bar._bar is not None
progress_bar._bar.update.assert_called()
def test_closes_when_finished(self, mock_manager: mock.MagicMock) -> None:
progress_bar = ui.ProgressBar("test")
mock_update_progress(progress_bar, 300)
with progress_bar:
pass
assert progress_bar._bar is not None
progress_bar._bar.close.assert_called_with(clear=True)
def test_creates_progress_bar_on_initial_update(
self, mock_manager: mock.MagicMock) -> None:
progress_bar = ui.ProgressBar("test_description")
mock_update_progress(progress_bar, 500)
mock_manager.counter.assert_called_with(desc="test_description",
leave=False,
total=500,
unit="frames")
def test_creates_new_bar_when_value_resets(
self, mock_manager: mock.MagicMock) -> None:
progress_bar = ui.ProgressBar("test")
mock_update_progress(progress_bar, 500)
mock_update_progress(progress_bar, 300)
assert mock_manager.counter.mock_calls == [
mock.call(desc="test", leave=False, total=500, unit="frames"),
mock.ANY,
mock.call(desc="test", leave=False, total=300, unit="frames"),
]