Rooba
/
tqdm
mirror of https://github.com/tqdm/tqdm.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

travis::flake8 compatible

This commit is contained in:
Casper da Costa-Luis 2015-06-15 10:44:41 +01:00 committed by Casper da Costa-Luis
parent 77c1d28159
commit a3a983dd3b
2 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tqdm/_tqdm.py
View File

@ -12,8 +12,8 @@ import sys
import time
__author__ = {"github.com/" : ["noamraph", "JackMc", "arkottke", "obiwanus",
"fordhurley", "kmike", "hadim", "casperdcl"]}
__author__ = {"github.com/": ["noamraph", "JackMc", "arkottke", "obiwanus",
"fordhurley", "kmike", "hadim", "casperdcl"]}
__all__ = ['tqdm', 'trange', 'format_interval', 'format_meter']
@ -53,8 +53,10 @@ def format_meter(n, total, elapsed, ncols=None, prefix=''):
bar_length = int(frac * N_BARS)
frac_bar_length = int((frac * N_BARS * 8) % 8)
try: unich = unichr
except: unich = chr
try:
unich = unichr
except:
unich = chr
bar = unich(0x2588)*bar_length
frac_bar = unich(0x2590 - frac_bar_length) if frac_bar_length else ' '
@ -85,7 +87,7 @@ def tqdm(iterable, desc=None, total=None, leave=False, file=sys.stderr,
Decorate an iterable object, returning an iterator which acts exactly
like the orignal iterable, but prints a dynamically updating
progressbar.
Parameters
----------
iterable : iterable
@ -109,7 +111,7 @@ def tqdm(iterable, desc=None, total=None, leave=False, file=sys.stderr,
Minimum progress update interval, in seconds [default: 0.5].
miniters : int, optional
Minimum progress update interval, in iterations [default: 1].
Returns
-------
out : decorated iterator.
@ -137,7 +139,7 @@ def tqdm(iterable, desc=None, total=None, leave=False, file=sys.stderr,
cur_t = time.time()
if cur_t - last_print_t >= mininterval:
sp.print_status(format_meter(
n, total, cur_t-start_t, ncols, prefix))
n, total, cur_t-start_t, ncols, prefix))
last_print_n = n
last_print_t = cur_t
@ -147,7 +149,8 @@ def tqdm(iterable, desc=None, total=None, leave=False, file=sys.stderr,
else:
if last_print_n < n:
cur_t = time.time()
sp.print_status(format_meter(n, total, cur_t-start_t, ncols, prefix))
sp.print_status(format_meter(
n, total, cur_t-start_t, ncols, prefix))
file.write('\n')

10
tqdm/tests/tests_tqdm.py
View File

@ -20,13 +20,15 @@ def test_format_interval():
def test_format_meter():
try: unich = unichr
except: unich = chr
try:
unich = unichr
except:
unich = chr
assert format_meter(0, 1000, 13) == \
" 0%| | 0/1000 [00:13<?, 0.00 it/s]"
assert format_meter(0, 1000, 13, ncols=70, prefix='desc: ') == \
"desc: 0%| | 0/1000 [00:13<?, 0.00 it/s]"
assert format_meter(0, 1000, 13, ncols=68, prefix='desc: ') == \
"desc: 0%| | 0/1000 [00:13<?, 0.00 it/s]"
assert format_meter(231, 1000, 392) == \
" 23%|" + unich(0x2588)*2 + unich(0x258e) + \
" | 231/1000 [06:32<21:44, 0.59 it/s]"