changing dense vector argmax unit test to be one line call vs 2

feynmanliang · Jun 9, 2015 · ac53c55 · ac53c55
1 parent d5b5423
commit ac53c55
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/linalg/VectorsSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/linalg/VectorsSuite.scala
@@ -65,8 +65,7 @@ class VectorsSuite extends FunSuite {
 
   test("dense argmax"){
     val vec = Vectors.dense(Array.empty[Double]).asInstanceOf[DenseVector]
-    val noMax = vec.argmax
-    assert(noMax === -1)
+    assert(vec.argmax === -1)
 
     val vec2 = Vectors.dense(arr).asInstanceOf[DenseVector]
     val max = vec2.argmax