Fix unnecessary sort method

prabeesh · May 11, 2015 · b8b5ef7 · b8b5ef7
1 parent d254be7
commit b8b5ef7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/pyspark/mllib/tests.py b/python/pyspark/mllib/tests.py
@@ -838,16 +838,17 @@ def test_load_vectors(self):
             [1.0, 2.0, 3.0],
             [1.0, 2.0, 3.0]
         ]
+        temp_dir = tempfile.mkdtemp()
+        load_vectors_path = os.path.join(temp_dir, "test_load_vectors")
         try:
-            temp_dir = tempfile.mkdtemp()
-            load_vectors_path = os.path.join(temp_dir, "test_load_vectors")
             self.sc.parallelize(data).saveAsTextFile(load_vectors_path)
             ret_rdd = MLUtils.loadVectors(self.sc, load_vectors_path)
             ret = ret_rdd.collect()
-            ret.sort()
             self.assertEqual(len(ret), 2)
             self.assertEqual(ret[0], DenseVector([1.0, 2.0, 3.0]))
             self.assertEqual(ret[1], DenseVector([1.0, 2.0, 3.0]))
+        except:
+            self.fail()
         finally:
             shutil.rmtree(load_vectors_path)