0001
0002
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017
0018 package org.apache.spark.ml.attribute;
0019
0020 import org.junit.Assert;
0021 import org.junit.Test;
0022
0023 public class JavaAttributeSuite {
0024
0025 @Test
0026 public void testAttributeType() {
0027 AttributeType numericType = AttributeType.Numeric();
0028 AttributeType nominalType = AttributeType.Nominal();
0029 AttributeType binaryType = AttributeType.Binary();
0030 Assert.assertEquals(numericType, NumericAttribute.defaultAttr().attrType());
0031 Assert.assertEquals(nominalType, NominalAttribute.defaultAttr().attrType());
0032 Assert.assertEquals(binaryType, BinaryAttribute.defaultAttr().attrType());
0033 }
0034
0035 @Test
0036 public void testNumericAttribute() {
0037 NumericAttribute attr = NumericAttribute.defaultAttr()
0038 .withName("age").withIndex(0).withMin(0.0).withMax(1.0).withStd(0.5).withSparsity(0.4);
0039 Assert.assertEquals(attr.withoutIndex(), Attribute.fromStructField(attr.toStructField()));
0040 }
0041
0042 @Test
0043 public void testNominalAttribute() {
0044 NominalAttribute attr = NominalAttribute.defaultAttr()
0045 .withName("size").withIndex(1).withValues("small", "medium", "large");
0046 Assert.assertEquals(attr.withoutIndex(), Attribute.fromStructField(attr.toStructField()));
0047 }
0048
0049 @Test
0050 public void testBinaryAttribute() {
0051 BinaryAttribute attr = BinaryAttribute.defaultAttr()
0052 .withName("clicked").withIndex(2).withValues("no", "yes");
0053 Assert.assertEquals(attr.withoutIndex(), Attribute.fromStructField(attr.toStructField()));
0054 }
0055 }