Call testPMMLWrite to actually verify PMML export

apache · Mar 30, 2018 · e2dbe15 · e2dbe15
1 parent 25d6f77
commit e2dbe15
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala
@@ -226,6 +226,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext with DefaultR
       val pmmlClusteringModel = pmml.getModels.get(0).asInstanceOf[ClusteringModel]
       assert(pmmlClusteringModel.getNumberOfClusters === clusterCenters.length)
     }
+    testPMMLWrite(sc, model, checkModel)
   }
 }