def test_defined_wont_fit(): progress_bar = ProgressBarYum(2000, 'file.iso', max_width=33) assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar) progress_bar.force_done = True assert 'file.iso | 0.0 B 00:00:00 ' == str(progress_bar) progress_bar = ProgressBarYum(2000, 'file.iso', max_width=30) assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar) progress_bar.force_done = True assert 'file.i | 0.0 B 00:00:00 ' == str(progress_bar) progress_bar = ProgressBarYum(2000, 'file.iso', max_width=20) assert ' 0% [] --- KiB/s | 0.0 B ' == str(progress_bar) progress_bar.force_done = True assert ' | 0.0 B 00:00:00 ' == str(progress_bar)
def test_undefined(): misc.terminal_width = lambda: 60 eta._NOW = lambda: 1411868721.5 progress_bar = ProgressBarYum(None, '', max_width=55) assert ' [ ] --- KiB/s | 0.0 B ' == str(progress_bar) assert ' [ ] --- KiB/s | 0.0 B ' == str(progress_bar) assert ' [ ] --- KiB/s | 0.0 B ' == str(progress_bar) eta._NOW = lambda: 1411868722.0 progress_bar.numerator = 10 assert ' [ ] --- KiB/s | 10 B ' == str(progress_bar) assert ' [ ] --- KiB/s | 10 B ' == str(progress_bar) eta._NOW = lambda: 1411868722.5 progress_bar.numerator = 100 assert ' [ ] 180 B/s | 100 B ' == str(progress_bar) eta._NOW = lambda: 1411868723.0 progress_bar.numerator = 1954727 assert ' [ ] 1.9 MiB/s | 1.9 MiB ' == str(progress_bar) assert ' [ ] 1.9 MiB/s | 1.9 MiB ' == str(progress_bar) eta._NOW = lambda: 1411868723.5 progress_bar.numerator = 4217583 assert ' [ ] 2.8 MiB/s | 4.0 MiB ' == str(progress_bar) eta._NOW = lambda: 1411868724.0 progress_bar.numerator = 6826725 assert ' [ ] 3.4 MiB/s | 6.5 MiB ' == str(progress_bar) eta._NOW = lambda: 1411868724.5 progress_bar.numerator = 8659265 assert ' [ ] 3.6 MiB/s | 8.3 MiB ' == str(progress_bar) eta._NOW = lambda: 1411868725.0 progress_bar.numerator = 8659265 assert ' [ ] 3.3 MiB/s | 8.3 MiB ' == str(progress_bar) eta._NOW = lambda: 1411868725.5 progress_bar.numerator = 21057295 assert ' [ ] 4.8 MiB/s | 20 MiB ' == str(progress_bar) eta._NOW = lambda: 1411868726.0 progress_bar.numerator = 65572196 assert ' [ ] 10 MiB/s | 62 MiB ' == str(progress_bar) progress_bar.force_done = True assert ' | 62 MiB 00:00:05 ' == str(progress_bar)
def progress_bar_yum(): files = { 'CentOS-7.0-1406-x86_64-DVD.iso': 10 if OPTIONS['--fast'] else 4148166656, 'CentOS-7.0-1406-x86_64-Everything.iso': 15 if OPTIONS['--fast'] else 7062159360, 'md5sum.txt': 5 if OPTIONS['--fast'] else 486, } for file_name, file_size in files.items(): bar = ProgressBarYum(0 if OPTIONS['--undefined'] else file_size, file_name) for i in range(0, file_size + 1, 2 if OPTIONS['--fast'] else int(file_size / 100.0)): bar.numerator = i print(bar, end='\r') sys.stdout.flush() time.sleep(0.25) bar.numerator = file_size bar.force_done = True print(bar)
def test_undefined(): misc.terminal_width = lambda: 60 eta._NOW = lambda: 1411868721.5 progress_bar = ProgressBarYum(None, '', max_width=55) assert ' [ ] --- KiB/s | 0.0 B ' == str( progress_bar) assert ' [ ] --- KiB/s | 0.0 B ' == str( progress_bar) assert ' [ ] --- KiB/s | 0.0 B ' == str( progress_bar) eta._NOW = lambda: 1411868722.0 progress_bar.numerator = 10 assert ' [ ] --- KiB/s | 10 B ' == str( progress_bar) assert ' [ ] --- KiB/s | 10 B ' == str( progress_bar) eta._NOW = lambda: 1411868722.5 progress_bar.numerator = 100 assert ' [ ] 180 B/s | 100 B ' == str( progress_bar) eta._NOW = lambda: 1411868723.0 progress_bar.numerator = 1954727 assert ' [ ] 1.9 MiB/s | 1.9 MiB ' == str( progress_bar) assert ' [ ] 1.9 MiB/s | 1.9 MiB ' == str( progress_bar) eta._NOW = lambda: 1411868723.5 progress_bar.numerator = 4217583 assert ' [ ] 2.8 MiB/s | 4.0 MiB ' == str( progress_bar) eta._NOW = lambda: 1411868724.0 progress_bar.numerator = 6826725 assert ' [ ] 3.4 MiB/s | 6.5 MiB ' == str( progress_bar) eta._NOW = lambda: 1411868724.5 progress_bar.numerator = 8659265 assert ' [ ] 3.6 MiB/s | 8.3 MiB ' == str( progress_bar) eta._NOW = lambda: 1411868725.0 progress_bar.numerator = 8659265 assert ' [ ] 3.3 MiB/s | 8.3 MiB ' == str( progress_bar) eta._NOW = lambda: 1411868725.5 progress_bar.numerator = 21057295 assert ' [ ] 4.8 MiB/s | 20 MiB ' == str( progress_bar) eta._NOW = lambda: 1411868726.0 progress_bar.numerator = 65572196 assert ' [ ] 10 MiB/s | 62 MiB ' == str( progress_bar) progress_bar.force_done = True assert ' | 62 MiB 00:00:05 ' == str( progress_bar)