Update rdd pipe tests for checkCode

(rather than mode)
tdas · Jul 7, 2015 · ab9a2e1 · ab9a2e1
1 parent eb4801c
commit ab9a2e1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/pyspark/tests.py b/python/pyspark/tests.py
@@ -879,13 +879,12 @@ def test_pipe_functions(self):
         rdd = self.sc.parallelize(data)
         with QuietTest(self.sc):
             self.assertEqual([], rdd.pipe('cc').collect())
-            self.assertRaises(Py4JJavaError, rdd.pipe('cc', mode='strict').collect)
+            self.assertRaises(Py4JJavaError, rdd.pipe('cc', checkCode=True).collect)
         result = rdd.pipe('cat').collect()
         result.sort()
         [self.assertEqual(x, y) for x, y in zip(data, result)]
-        self.assertRaises(Py4JJavaError, rdd.pipe('grep 4', mode='strict').collect)
+        self.assertRaises(Py4JJavaError, rdd.pipe('grep 4', checkCode=True).collect)
         self.assertEqual([], rdd.pipe('grep 4').collect())
-        self.assertEqual([], rdd.pipe('grep 4', mode='grep').collect())
 
 
 class ProfilerTests(PySparkTestCase):