0001
0002
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017
0018 package org.apache.spark.ml.param;
0019
0020 import java.util.Arrays;
0021
0022 import org.junit.Assert;
0023 import org.junit.Test;
0024
0025
0026
0027
0028 public class JavaParamsSuite {
0029
0030 @Test
0031 public void testParams() {
0032 JavaTestParams testParams = new JavaTestParams();
0033 Assert.assertEquals(1, testParams.getMyIntParam());
0034 testParams.setMyIntParam(2).setMyDoubleParam(0.4).setMyStringParam("a");
0035 Assert.assertEquals(0.4, testParams.getMyDoubleParam(), 0.0);
0036 Assert.assertEquals("a", testParams.getMyStringParam());
0037 Assert.assertArrayEquals(testParams.getMyDoubleArrayParam(), new double[]{1.0, 2.0}, 0.0);
0038 }
0039
0040 @Test
0041 public void testParamValidate() {
0042 ParamValidators.gt(1.0);
0043 ParamValidators.gtEq(1.0);
0044 ParamValidators.lt(1.0);
0045 ParamValidators.ltEq(1.0);
0046 ParamValidators.inRange(0, 1, true, false);
0047 ParamValidators.inRange(0, 1);
0048 ParamValidators.inArray(Arrays.asList(0, 1, 3));
0049 ParamValidators.inArray(Arrays.asList("a", "b"));
0050 }
0051 }