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_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