Add test for covariance matrix

apache · Feb 10, 2015 · e579041 · e579041
1 parent 5cb370b
commit e579041
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/clustering/GaussianMixtureSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/clustering/GaussianMixtureSuite.scala
@@ -100,6 +100,7 @@ class GaussianMixtureSuite extends FunSuite with MLlibTestSparkContext {
       val gmm = new GaussianMixture().setK(1).setSeed(seed).run(data)
       assert(gmm.weights(0) ~== Ew absTol 1E-5)
       assert(gmm.gaussians(0).mu ~== Emu absTol 1E-5)
+      assert(gmm.gaussians(0).sigma ~== Esigma absTol 1E-5)
     }
   }