Rename a variable in BisectingKMeansModelSuite

apache · Oct 29, 2015 · a876ba2 · a876ba2
1 parent 73e2c7a
commit a876ba2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/clustering/BisectingKMeansModelSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/clustering/BisectingKMeansModelSuite.scala
@@ -104,25 +104,25 @@ class BisectingKMeansModelSuite
   }
 
   test("clustering should be done correctly") {
-    for (numClusters <- Array(9, 19)) {
-      val app = new BisectingKMeans().setK(numClusters).setSeed(1)
+    for (k <- Array(9, 19)) {
+      val app = new BisectingKMeans().setK(k).setSeed(1)
       val localData = (1 to 19).toSeq.map { i =>
-        val label = i % numClusters
-        val sparseVector = Vectors.sparse(numClusters, Seq((label, label.toDouble)))
+        val label = i % k
+        val sparseVector = Vectors.sparse(k, Seq((label, label.toDouble)))
         val denseVector = Vectors.fromBreeze(sparseVector.toBreeze.toDenseVector)
         (label, denseVector, sparseVector)
       }
 
       // dense version
       val denseData = sc.parallelize(localData.map(_._2), 2)
       val denseModel = app.run(denseData)
-      assert(denseModel.getCenters.length === numClusters)
+      assert(denseModel.getCenters.length === k)
       assert(denseModel.getClusters.forall(_.cost == 0.0))
 
       // sparse version
       val sparseData = sc.parallelize(localData.map(_._3), 2)
       val sparseModel = app.run(sparseData)
-      assert(sparseModel.getCenters.length === numClusters)
+      assert(sparseModel.getCenters.length === k)
       assert(sparseModel.getClusters.forall(_.cost == 0.0))
     }
   }