Condense test

Author: schintap

python/pyspark/tests/test_rdd.py

@@ -170,12 +170,12 @@ def test_zip_chaining(self):
 
     def test_union_pair_rdd(self):
         # Regression test for SPARK-31788
-        rdd1 = self.sc.parallelize([1, 2])
-        rdd2 = self.sc.parallelize([3, 4])
-        pair_rdd = rdd1.zip(rdd2)
-        expected = [(1, 3), (2, 4), (1, 3), (2, 4)]
-        actual = self.sc.union([pair_rdd, pair_rdd]).collect()
-        self.assertEqual(expected, actual)
+        rdd = self.sc.parallelize([1, 2])
+        pair_rdd = rdd.zip(rdd)
+        self.assertEqual(
+            self.sc.union([pair_rdd, pair_rdd]).collect(),
+            [((1, 1), (2, 2)), ((1, 1), (2, 2))]
+        )
 
     def test_deleting_input_files(self):
         # Regression test for SPARK-1025