diff options
Diffstat (limited to 'mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala')
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala b/mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala index 0aa774b660..8e5365af84 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala @@ -82,6 +82,7 @@ trait DefaultReadWriteTest extends TempDirectory { self: Suite => * - Explicitly set Params, and train model * - Test save/load using [[testDefaultReadWrite()]] on Estimator and Model * - Check Params on Estimator and Model + * - Compare model data * * This requires that the [[Estimator]] and [[Model]] share the same set of [[Param]]s. * @param estimator Estimator to test @@ -117,6 +118,8 @@ trait DefaultReadWriteTest extends TempDirectory { self: Suite => val param = model.getParam(p) assert(model.get(param).get === model2.get(param).get) } + + checkModelData(model, model2) } } |