def test_balance_data():
# pylint: disable=missing-docstring
data = SList.init(int, 23)
size = data.length()
res = data.balance()
exp = SList(range(0, size))
assert res == exp
def _test3():
if SEQ:
data = DSList.init(_vrand, SIZE)
else:
data = DPList.init(_vrand, SIZE)
res1 = test_timing(_vavg, data, "vector average")
res2 = test_timing(_wrapped_vavg, data, "vector average [_run]")
assert res1 == res2
def test_distribute_data():
# pylint: disable=missing-docstring
data = SList.init(int, 42)
size = data.length()
distr = Distribution([size])
res = data.distribute(distr)
exp = SList(range(0, size))
assert res == exp
def _test2():
if SEQ:
input2 = DSList.init(lambda i: random.randint(0, 9) <= 4, SIZE)
else:
input2 = DPList.init(lambda i: random.randint(0, 9) <= 4, SIZE)
res1 = test_timing(_test_bool_direct, input2, "map/reduce bool")
res2 = test_timing(_test_bool_mr, input2, "map_reduce bool")
res3 = test_timing(_test_bool_optimized, input2, "map/reduce bool[_opt]", _opt, _run)
assert res1 == res3
assert res1 == res2
def _test1():
if SEQ:
input1 = DSList.init(lambda num: random.randint(0, 1000), SIZE)
else:
input1 = DPList.init(lambda num: random.randint(0, 1000), SIZE)
res1 = test_timing(_test_mmr_direct, input1, "map/map/reduce")
res2 = test_timing(_test_mmr_direct, input1, "map/reduce[hc]")
res3 = test_timing(_test_mmr_run, input1, "map/map/reduce[_opt]")
res4 = test_timing(_test_mmr_optimized, input1, "map/map/reduce[_opt/_run]", _opt, _run)
assert res1 == res2 and res1 == res3 and res1 == res4
def _vrand(_):
return DSList.init(_f_rand, DIM)
def _test2():
if SEQ:
input2 = DSList.init(lambda i: random.randint(0, 9) <= 4, SIZE)
else:
input2 = DPList.init(lambda i: random.randint(0, 9) <= 4, SIZE)
res1 = test_timing(_test_bool_direct, input2, "map/reduce bool")
res2 = test_timing(_test_bool_mr, input2, "map_reduce bool")
res3 = test_timing(_test_bool_optimized, input2, "map/reduce bool[_opt]", _opt, _run)
assert res1 == res3
assert res1 == res2
DIM = 10
VZERO = DSList.init(lambda i: 0.0, DIM)
def _f_rand(_):
return random.randint(0, 1000)
def _vrand(_):
return DSList.init(_f_rand, DIM)
def _vnsum(lst):
return lst.map(normalize).reduce(vadd, VZERO)
def _vavg(lst):
def test_permute_swap():
# pylint: disable=missing-docstring
exp = SList.init(str, 43)
size = exp.length()
res = exp.permute(swap(size)).permute(swap(size))
assert exp == res
def test_permute_idt():
# pylint: disable=missing-docstring
input_list = SList.init(str, 37)
exp = input_list
res = input_list.permute(fun.idt)
assert exp == res
def test_scatter_range_data():
# pylint: disable=missing-docstring
data = SList.init(float, 111)
res = data.scatter_range(range(23, 27))
exp = SList(range(23, 27)).map(float)
assert res == exp
def test_scatter_data():
# pylint: disable=missing-docstring
data = SList.init(str, 13)
res = data.scatter(0)
exp = data
assert res == exp
def test_gather_data():
# pylint: disable=missing-docstring
data = SList.init(float, 17)
res = data.gather(0)
exp = SList(range(0, len(data)))
assert res == exp
