From c4652ea05e05b46480b3399255f7a7bb299a5b2d Mon Sep 17 00:00:00 2001 From: Davies Liu Date: Tue, 19 Aug 2014 00:14:44 -0700 Subject: [PATCH] add more test cases --- python/pyspark/tests.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/python/pyspark/tests.py b/python/pyspark/tests.py index 305c82e5d59d3..43ea3b7b19e8f 100644 --- a/python/pyspark/tests.py +++ b/python/pyspark/tests.py @@ -349,6 +349,11 @@ def test_zip_with_different_number_of_items(self): # different number of items in one pair b = self.sc.parallelize(range(100, 106), 2) self.assertRaises(Exception, lambda: a.zip(b).count()) + # same total number of items, but different distributions + a = self.sc.parallelize([2, 3], 2).flatMap(range) + b = self.sc.parallelize([3, 2], 2).flatMap(range) + self.assertEquals(a.count(), b.count()) + self.assertRaises(Exception, lambda: a.zip(b).count()) class TestIO(PySparkTestCase):