def test_count_partitions():
my_rdd = Context().parallelize([1, 2, 3], 2)
print(my_rdd.collect())
my_rdd.foreach(print)
assert my_rdd.count() == 3
def test_foreach():
my_rdd = Context().parallelize([1, 2, 3])
a = []
my_rdd.foreach(lambda x: a.append(x))
assert len(a) == 3
