def test_take_partitions():
"""The real test here is that only the first two partitions should be
computed and not the third one. Shown in debug logs."""
my_rdd = Context().parallelize([4, 7, 2], 3)
assert my_rdd.take(2)[1] == 7
def test_take():
my_rdd = Context().parallelize([4, 7, 2])
assert my_rdd.take(2)[1] == 7
