@Test
public void transformPredictDataToTrainDataTest() {

    KMeansPredictModelData predictModelData = new KMeansModelDataConverter().load(modelRows);
    Assert.assertEquals(predictModelData.params.k, 2);
    Assert.assertEquals(predictModelData.params.vectorSize, 3);
    Assert.assertEquals(predictModelData.params.vectorColName, "Y");
    Assert.assertNull(predictModelData.params.longtitudeColName);
    Assert.assertNull(predictModelData.params.latitudeColName);
    Assert.assertEquals(predictModelData.params.distanceType, HasKMeansWithHaversineDistanceType.DistanceType.EUCLIDEAN);
    Assert.assertEquals(predictModelData.centroids.getVectors(), new DenseMatrix(3, 2, new double[]{9.1, 9.1, 9.1, 0.1, 0.1, 0.1}));
}