add more test cases

xiliu82 · Aug 19, 2014 · c4652ea · c4652ea
1 parent 6d05fc8
commit c4652ea
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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):