def test_parallel_map():
def fn(n):
return n * 2
input = [1, 2, 3, 4]
result = parallel_map(4, fn, input)
assert frozenset(result) == frozenset(map(fn, input))
def test_parallel_map_multiargs():
def fn(n, m):
return n * m
input1 = [1, 2, 3, 4]
input2 = [5, 6, 7, 8]
result = parallel_map(4, fn, input1, input2)
assert frozenset(result) == frozenset(map(fn, input1, input2))
def test_race_condition(fx_stages, fx_feed):
stage, _ = fx_stages
with stage:
stage.feeds["test"] = fx_feed
result = parallel_map(10, functools.partial(apply_timestamp, stage, "test"), map(timestamp, range(10)))
for _ in result:
pass
with stage:
updated_at = stage.feeds["test"].entries[0].read.updated_at
assert updated_at == timestamp(9)
def test_race_condition(fx_stages, fx_feed):
stage, _ = fx_stages
with stage:
stage.feeds['test'] = fx_feed
result = parallel_map(10, functools.partial(apply_timestamp, stage,
'test'),
map(timestamp, range(10)))
for _ in result:
pass
with stage:
updated_at = stage.feeds['test'].entries[0].read.updated_at
assert updated_at == timestamp(9)
def test_parallel_map_errors_in_loop():
input = [0, 1, 2, 3]
result = parallel_map(4, lambda n: 1 // n, input)
it = iter(result)
results = []
while True:
try:
results.append(next(it))
except ZeroDivisionError:
continue
except StopIteration:
break
assert len(results) == 3
def test_parallel_map_errors():
input = [1, 2, 3, 0]
result = parallel_map(4, lambda n: 1 // n, input)
it = iter(result)
# Although coudn't find why, pytest.raises() context manager
# seems not to work on IronPython.
try:
next(it)
next(it)
next(it)
next(it)
except ZeroDivisionError:
pass
else:
assert False, 'expected ZeroDivisionError, but it was not raised'
def test_parallel_map_multiargs():
input1 = [1, 2, 3, 4]
input2 = [5, 6, 7, 8]
fn = lambda n, m: n * m
result = parallel_map(4, fn, input1, input2)
assert frozenset(result) == frozenset(map(fn, input1, input2))
def test_parallel_map():
input = [1, 2, 3, 4]
fn = lambda n: n * 2
result = parallel_map(4, fn, input)
assert frozenset(result) == frozenset(map(fn, input))
def test_parallel_map_multiargs():
input1 = [1, 2, 3, 4]
input2 = [5, 6, 7, 8]
fn = lambda n, m: n * m
result = parallel_map(4, fn, input1, input2)
assert frozenset(result) == frozenset(map(fn, input1, input2))
def test_parallel_map():
input = [1, 2, 3, 4]
fn = lambda n: n * 2
result = parallel_map(4, fn, input)
assert frozenset(result) == frozenset(map(fn, input))
