def test_progress_shown_if_eta_above_threshold(self):
progress_mock = self.ProgressMock(20)
progress_bar = ProgressBarMock()
auto_progress = AutoProgressBar(progress_bar, min_eta=10.)
for _ in range(10):
auto_progress.update(progress_mock)
assert progress_bar.n_update_calls >= 10
def test_progress_not_shown_if_eta_below_threshold(self):
progress_mock = self.ProgressMock(0.2)
progress_bar = ProgressBarMock()
auto_progress = AutoProgressBar(progress_bar, min_eta=10.0)
for _ in range(10):
auto_progress.update(progress_mock)
assert progress_bar.n_update_calls == 0
def test_progress_shown_if_eta_above_threshold(self):
progress_mock = self.ProgressMock(20)
progress_bar = ProgressBarMock()
auto_progress = AutoProgressBar(progress_bar, min_eta=10.)
for _ in range(10):
auto_progress.update(progress_mock)
assert progress_bar.n_update_calls >= 10
def test_progress_shown_when_finished(self):
progress_mock = self.ProgressMock(0.2)
progress_bar = ProgressBarMock()
auto_progress = AutoProgressBar(progress_bar, min_eta=10.)
auto_progress.update(progress_mock)
assert progress_bar.n_update_calls == 0
progress_mock.finished = True
auto_progress.update(progress_mock)
assert progress_bar.n_update_calls >= 1
def test_progress_shown_when_finished(self):
progress_mock = self.ProgressMock(0.2)
progress_bar = ProgressBarMock()
auto_progress = AutoProgressBar(progress_bar, min_eta=10.)
auto_progress.update(progress_mock)
assert progress_bar.n_update_calls == 0
progress_mock.finished = True
auto_progress.update(progress_mock)
assert progress_bar.n_update_calls >= 1
def test_close(self):
progress_bar = ProgressBarMock()
auto_progress = AutoProgressBar(progress_bar)
auto_progress.close()
assert progress_bar.closed
