mirror of https://github.com/tqdm/tqdm.git
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
"""
|
|
Tests for `tqdm.contrib.concurrent`.
|
|
"""
|
|
from pytest import mark, warns
|
|
|
|
from tqdm.contrib.concurrent import thread_map, process_map
|
|
from .tests_tqdm import importorskip, skip, StringIO, closing, TqdmWarning
|
|
|
|
|
|
def incr(x):
|
|
"""Dummy function"""
|
|
return x + 1
|
|
|
|
|
|
def test_thread_map():
|
|
"""Test contrib.concurrent.thread_map"""
|
|
with closing(StringIO()) as our_file:
|
|
a = range(9)
|
|
b = [i + 1 for i in a]
|
|
try:
|
|
assert thread_map(lambda x: x + 1, a, file=our_file) == b
|
|
except ImportError as err:
|
|
skip(str(err))
|
|
assert thread_map(incr, a, file=our_file) == b
|
|
|
|
|
|
def test_process_map():
|
|
"""Test contrib.concurrent.process_map"""
|
|
with closing(StringIO()) as our_file:
|
|
a = range(9)
|
|
b = [i + 1 for i in a]
|
|
try:
|
|
assert process_map(incr, a, file=our_file) == b
|
|
except ImportError as err:
|
|
skip(str(err))
|
|
|
|
|
|
@mark.parametrize("iterables,should_warn", [([], False), (['x'], False), ([()], False),
|
|
(['x', ()], False), (['x' * 1001], True),
|
|
(['x' * 100, ('x',) * 1001], True)])
|
|
def test_chunksize_warning(iterables, should_warn):
|
|
"""Test contrib.concurrent.process_map chunksize warnings"""
|
|
patch = importorskip("unittest.mock").patch
|
|
with patch('tqdm.contrib.concurrent._executor_map'):
|
|
if should_warn:
|
|
warns(TqdmWarning, process_map, incr, *iterables)
|
|
else:
|
|
process_map(incr, *iterables)
|