From a38527fce73ce4744ca10e5376c4b6fe2c2ff93b Mon Sep 17 00:00:00 2001 From: Jyotiska NK Date: Wed, 19 Mar 2014 10:56:02 +0530 Subject: [PATCH] Added doctest for map function in rdd.py Doctest added for map in rdd.py --- python/pyspark/rdd.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/pyspark/rdd.py b/python/pyspark/rdd.py index ae09dbff02a3..ca2dc119a461 100644 --- a/python/pyspark/rdd.py +++ b/python/pyspark/rdd.py @@ -180,6 +180,10 @@ def getCheckpointFile(self): def map(self, f, preservesPartitioning=False): """ Return a new RDD by applying a function to each element of this RDD. + + >>> rdd = sc.parallelize(["b", "a", "c"]) + >>> sorted(rdd.map(lambda x: (x, 1)).collect()) + [('a', 1), ('b', 1), ('c', 1)] """ def func(split, iterator): return imap(f, iterator) return PipelinedRDD(self, func, preservesPartitioning)