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