liting and minor tidy

This commit is contained in:
Casper da Costa-Luis 2019-11-08 23:20:43 +00:00
parent 213f30288c
commit 166e1bc5dc
No known key found for this signature in database
GPG Key ID: 986B408043AE090D
2 changed files with 9 additions and 10 deletions

View File

@ -222,11 +222,11 @@ def test_manual_overhead():
def worker(total, blocking=True):
def incr_bar(x):
with closing(StringIO()) as our_file:
with tqdm(total=x, file=our_file,
lock_args=None if blocking else (False,),
miniters=1, mininterval=0, maxinterval=0) as t:
for i in range(total):
t.update()
for _ in trange(
total, file=our_file,
lock_args=None if blocking else (False,),
miniters=1, mininterval=0, maxinterval=0):
pass
return x + 1
return incr_bar
@ -243,7 +243,7 @@ def test_lock_args():
import sys
total = 8
subtotal = 100
subtotal = 1000
tqdm.set_lock(RLock())
with ThreadPoolExecutor(total) as pool:

View File

@ -1,5 +1,5 @@
from __future__ import division
from tqdm import tqdm, TMonitor
from tqdm import tqdm, trange, TMonitor
from tests_tqdm import with_setup, pretest, posttest, SkipTest, \
StringIO, closing
from tests_tqdm import DiscreteTimer, cpu_timify
@ -43,9 +43,8 @@ def incr(x):
def incr_bar(x):
with closing(StringIO()) as our_file:
with tqdm(total=x, lock_args=(False,), file=our_file) as t:
for i in range(x):
t.update()
for _ in trange(x, lock_args=(False,), file=our_file):
pass
return incr(x)