refactor

apache · Sep 11, 2014 · 4bc1f04 · 4bc1f04
1 parent b20ab3a
commit 4bc1f04
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/python/pyspark/tests.py b/python/pyspark/tests.py
@@ -205,15 +205,11 @@ def __getattr__(self, item):
         getter2 = ser.loads(ser.dumps(getter))
         self.assertEqual(getter(d), getter2(d))
 
-
-# Regression test for SPARK-3415
-class CloudPickleTest(unittest.TestCase):
+    # Regression test for SPARK-3415
     def test_pickling_file_handles(self):
-        from pyspark.cloudpickle import dumps
-        from StringIO import StringIO
-        from pickle import load
+        ser = CloudPickleSerializer()
         out1 = sys.stderr
-        out2 = load(StringIO(dumps(out1)))
+        out2 = ser.loads(ser.dumps(out1))
         self.assertEquals(out1, out2)