def test_cache():
# this crashes in version 0.2.28
lines = Context().textFile('tests/*textFil*.py')
lines = lines.map(lambda l: '-'+l).cache()
print(len(lines.collect()))
lines = lines.map(lambda l: '+'+l)
lines = lines.map(lambda l: '-'+l).cache()
lines = lines.collect()
print(lines)
assert '-+-from pysparkling import Context' in lines
def test_cache():
# this crashes in version 0.2.28
lines = Context().textFile('tests/*textFil*.py')
lines = lines.map(lambda l: '-' + l).cache()
print(len(lines.collect()))
lines = lines.map(lambda l: '+' + l)
lines = lines.map(lambda l: '-' + l).cache()
lines = lines.collect()
print(lines)
assert '-+-from pysparkling import Context' in lines
def test_lazy_execution_threadpool():
def indent_line(l):
return '--- ' + l
with futures.ThreadPoolExecutor(4) as p:
r = Context(pool=p).textFile('tests/test_multiprocessing.py')
r = r.map(indent_line).cache()
r.collect()
r = r.map(indent_line)
r = r.collect()
# ThreadPool is not lazy although it returns generators.
print(r)
assert '--- --- from pysparkling import Context' in r
def test_lazy_execution_threadpool():
def indent_line(l):
return '--- '+l
with futures.ThreadPoolExecutor(4) as p:
r = Context(pool=p).textFile('tests/test_multiprocessing.py')
r = r.map(indent_line).cache()
r.collect()
r = r.map(indent_line)
r = r.collect()
# ThreadPool is not lazy although it returns generators.
print(r)
assert '--- --- from pysparkling import Context' in r
def test_lazy_execution_processpool():
with futures.ProcessPoolExecutor(4) as p:
r = Context(
pool=p,
serializer=dill.dumps,
deserializer=dill.loads,
).textFile('tests/test_multiprocessing.py')
r = r.map(indent_line).cache()
r.collect()
r = r.map(indent_line)
r = r.collect()
# ProcessPool is not lazy although it returns generators.
print(r)
assert '--- --- from pysparkling import Context' in r
def test_lazy_execution():
r = Context().textFile('tests/test_multiprocessing.py')
r = r.map(indent_line)
exec_before_collect = INDENT_WAS_EXECUTED
# at this point, no map() or foreach() should have been executed
r.collect()
exec_after_collect = INDENT_WAS_EXECUTED
assert not exec_before_collect and exec_after_collect
def test_lazy_execution_processpool():
def indent_line(l):
return '--- ' + l
with futures.ProcessPoolExecutor(4) as p:
r = Context(
pool=p,
serializer=cloudpickle.dumps,
deserializer=pickle.loads,
).textFile('tests/test_multiprocessing.py') # .take(10)
print(r.collect())
r = r.map(indent_line)
print(r.collect())
r = r.cache()
print(r.collect())
r = r.map(indent_line)
r = r.collect()
# ProcessPool is not lazy although it returns generators.
print(r)
assert '--- --- from pysparkling import Context' in r
def test_cache():
my_rdd = Context().parallelize([1, 2, 3, 4], 2)
my_rdd = my_rdd.map(lambda x: x*x).cache()
print('no exec until here')
print(my_rdd.first())
print('executed map on first partition only')
print(my_rdd.collect())
print('now map() was executed on all partitions and should '
'not be executed again')
print(my_rdd.collect())
assert len(my_rdd.collect()) == 4 and 16 in my_rdd.collect()
def test_lazy_execution_processpool():
def indent_line(l):
return '--- '+l
with futures.ProcessPoolExecutor(4) as p:
r = Context(
pool=p,
serializer=cloudpickle.dumps,
deserializer=pickle.loads,
).textFile('tests/test_multiprocessing.py') # .take(10)
print(r.collect())
r = r.map(indent_line)
print(r.collect())
r = r.cache()
print(r.collect())
r = r.map(indent_line)
r = r.collect()
# ProcessPool is not lazy although it returns generators.
print(r)
assert '--- --- from pysparkling import Context' in r
def test_lazy_execution():
class I(object):
def __init__(self):
self.executed = False
def indent_line(self, l):
# global indent_was_executed
self.executed = True
return '--- ' + l
r = Context().textFile('tests/test_multiprocessing.py')
i = I()
r = r.map(i.indent_line)
exec_before_collect = i.executed
# at this point, no map() or foreach() should have been executed
r = r.map(i.indent_line).cache()
print(r.collect())
r = r.map(i.indent_line)
r.collect()
exec_after_collect = i.executed
print((exec_before_collect, exec_after_collect))
assert not exec_before_collect and exec_after_collect
def test_processpool_distributed_cache():
with futures.ProcessPoolExecutor(4) as p:
r = Context(
pool=p,
serializer=cloudpickle.dumps,
deserializer=pickle.loads,
).parallelize(range(3), 3)
r = r.map(lambda _: time.sleep(0.1)).cache()
r.collect()
time_start = time.time()
print(r.collect())
time_end = time.time()
assert time_end - time_start < 0.3
def test_processpool_distributed_cache():
with futures.ProcessPoolExecutor(4) as p:
r = Context(
pool=p,
serializer=cloudpickle.dumps,
deserializer=pickle.loads,
).parallelize(range(3), 3)
r = r.map(lambda _: time.sleep(0.1)).cache()
r.collect()
time_start = time.time()
print(r.collect())
time_end = time.time()
assert time_end - time_start < 0.3
def test_lazy_execution():
class I(object):
def __init__(self):
self.executed = False
def indent_line(self, l):
# global indent_was_executed
self.executed = True
return '--- '+l
r = Context().textFile('tests/test_multiprocessing.py')
i = I()
r = r.map(i.indent_line)
exec_before_collect = i.executed
# at this point, no map() or foreach() should have been executed
r = r.map(i.indent_line).cache()
print(r.collect())
r = r.map(i.indent_line)
r.collect()
exec_after_collect = i.executed
print((exec_before_collect, exec_after_collect))
assert not exec_before_collect and exec_after_collect
def test_concurrent():
with futures.ThreadPoolExecutor(4) as p:
my_rdd = Context(pool=p).parallelize([1, 3, 4])
r = my_rdd.map(math.sqrt).collect()
print(r)
assert 2 in r
def test_concurrent():
with futures.ThreadPoolExecutor(4) as p:
my_rdd = Context(pool=p).parallelize([1, 3, 4])
r = my_rdd.map(math.sqrt).collect()
print(r)
assert 2 in r
