fix equal of matrix

Author: davies

mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala

@@ -17,6 +17,8 @@
 
 package org.apache.spark.mllib.linalg
 
+import java.util
+
 import breeze.linalg.{Matrix => BM, DenseMatrix => BDM}
 
 /**
@@ -66,8 +68,8 @@ class DenseMatrix(val numRows: Int, val numCols: Int, val values: Array[Double])
   private[mllib] override def toBreeze: BM[Double] = new BDM[Double](numRows, numCols, values)
 
   override def equals(o: Any) = o match {
-    case that: DenseMatrix =>
-      that.numRows == numRows && that.numCols == numCols && values.equals(that.values)
+    case m: DenseMatrix =>
+      m.numRows == numRows && m.numCols == numCols && util.Arrays.equals(toArray, m.toArray)
     case _ => false
   }
 }

mllib/src/test/scala/org/apache/spark/mllib/api/python/PythonMLLibAPISuite.scala

@@ -70,7 +70,7 @@ class PythonMLLibAPISuite extends FunSuite {
     val values = Array[Double](0, 1.2, 3, 4.56, 7, 8)
     val matrix = Matrices.dense(2, 3, values)
     val nm = SerDe.loads(SerDe.dumps(matrix)).asInstanceOf[DenseMatrix]
-    assert(matrix == nm)
+    assert(matrix === nm)
 
     // Test conversion for empty matrix
     val empty = Array[Double]()