From 64561e4e503eafb958f6769383ba3b37edbe5fa2 Mon Sep 17 00:00:00 2001 From: Davies Liu Date: Fri, 10 Oct 2014 22:47:46 -0700 Subject: [PATCH] fix tests --- python/pyspark/streaming/tests.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/python/pyspark/streaming/tests.py b/python/pyspark/streaming/tests.py index 0e5c1a3b3c2ad..a8d876d0fa3b3 100644 --- a/python/pyspark/streaming/tests.py +++ b/python/pyspark/streaming/tests.py @@ -487,9 +487,8 @@ def test_get_or_create(self): inputd = tempfile.mkdtemp() outputd = tempfile.mkdtemp() + "/" - def updater(it): - for k, vs, s in it: - yield (k, sum(vs, s or 0)) + def updater(vs, s): + return sum(vs, s or 0) def setup(): conf = SparkConf().set("spark.default.parallelism", 1)