public class TestVectorizationContext {
@Test
public void testVectorExpressionDescriptor() {
VectorUDFUnixTimeStampLong v1 = new VectorUDFUnixTimeStampLong();
VectorExpressionDescriptor.Builder builder1 = new VectorExpressionDescriptor.Builder();
VectorExpressionDescriptor.Descriptor d1 = builder1.setMode(VectorExpressionDescriptor.Mode.PROJECTION)
.setNumArguments(1).setArgumentTypes(VectorExpressionDescriptor.ArgumentType.LONG)
.setInputExpressionTypes(VectorExpressionDescriptor.InputExpressionType.COLUMN).build();
Assert.assertEquals(d1, v1.getDescriptor());
VectorExpressionDescriptor.Builder builder2 = new VectorExpressionDescriptor.Builder();
VectorExpressionDescriptor.Descriptor d2 = builder2.setMode(VectorExpressionDescriptor.Mode.FILTER)
.setNumArguments(2).setArgumentTypes(VectorExpressionDescriptor.ArgumentType.LONG,
VectorExpressionDescriptor.ArgumentType.DOUBLE).setInputExpressionTypes(