Search in sources :

Example 31 with SingleInputSemanticProperties

use of org.apache.flink.api.common.operators.SingleInputSemanticProperties in project flink by apache.

the class SemanticPropUtilTest method testForwardedInvalidString.

@Test(expected = InvalidSemanticAnnotationException.class)
public void testForwardedInvalidString() {
    String[] forwardedFields = { "notValid" };
    SingleInputSemanticProperties sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, threeIntTupleType, threeIntTupleType);
}
Also used : SingleInputSemanticProperties(org.apache.flink.api.common.operators.SingleInputSemanticProperties) Test(org.junit.Test)

Example 32 with SingleInputSemanticProperties

use of org.apache.flink.api.common.operators.SingleInputSemanticProperties in project flink by apache.

the class SemanticPropUtilTest method testForwardedNoArrowOneStringInvalidDelimiter.

@Test(expected = InvalidSemanticAnnotationException.class)
public void testForwardedNoArrowOneStringInvalidDelimiter() {
    String[] forwardedFields = { "f2,f3,f0" };
    SingleInputSemanticProperties sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, fiveIntTupleType, fiveIntTupleType);
}
Also used : SingleInputSemanticProperties(org.apache.flink.api.common.operators.SingleInputSemanticProperties) Test(org.junit.Test)

Example 33 with SingleInputSemanticProperties

use of org.apache.flink.api.common.operators.SingleInputSemanticProperties in project flink by apache.

the class SemanticPropUtilTest method testReadFieldsOneString.

@Test
public void testReadFieldsOneString() {
    String[] readFields = { "f1;f2" };
    SingleInputSemanticProperties sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, null, null, readFields, threeIntTupleType, threeIntTupleType);
    FieldSet fs = sp.getReadFields(0);
    assertTrue(fs.size() == 2);
    assertTrue(fs.contains(2));
    assertTrue(fs.contains(1));
    readFields[0] = "f1;f2;";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, null, null, readFields, threeIntTupleType, threeIntTupleType);
    fs = sp.getReadFields(0);
    assertTrue(fs.size() == 2);
    assertTrue(fs.contains(2));
    assertTrue(fs.contains(1));
}
Also used : FieldSet(org.apache.flink.api.common.operators.util.FieldSet) SingleInputSemanticProperties(org.apache.flink.api.common.operators.SingleInputSemanticProperties) Test(org.junit.Test)

Example 34 with SingleInputSemanticProperties

use of org.apache.flink.api.common.operators.SingleInputSemanticProperties in project flink by apache.

the class SemanticPropUtilTest method testForwardedNestedTuples.

@Test
public void testForwardedNestedTuples() {
    String[] forwardedFields = { "f0->f0.f0; f1->f0.f1; f2->f0.f2" };
    SingleInputSemanticProperties sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, threeIntTupleType, nestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 0).contains(0));
    assertTrue(sp.getForwardingTargetFields(0, 1).contains(1));
    assertTrue(sp.getForwardingTargetFields(0, 2).contains(2));
    forwardedFields[0] = "f0.f0->f1.f0.f2; f0.f1->f2; f2->f1.f2; f1->f0";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, nestedTupleType, deepNestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 0).contains(3));
    assertTrue(sp.getForwardingTargetFields(0, 1).contains(6));
    assertTrue(sp.getForwardingTargetFields(0, 4).contains(5));
    assertTrue(sp.getForwardingTargetFields(0, 3).contains(0));
    forwardedFields[0] = "0.0->1.0.2; 0.1->2; 2->1.2; 1->0";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, nestedTupleType, deepNestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 0).contains(3));
    assertTrue(sp.getForwardingTargetFields(0, 1).contains(6));
    assertTrue(sp.getForwardingTargetFields(0, 4).contains(5));
    assertTrue(sp.getForwardingTargetFields(0, 3).contains(0));
    forwardedFields[0] = "f1.f0.*->f0.*; f0->f2";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, deepNestedTupleType, nestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 1).contains(0));
    assertTrue(sp.getForwardingTargetFields(0, 2).contains(1));
    assertTrue(sp.getForwardingTargetFields(0, 3).contains(2));
    assertTrue(sp.getForwardingTargetFields(0, 0).contains(4));
    forwardedFields[0] = "1.0.*->0.*; 0->2";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, deepNestedTupleType, nestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 1).contains(0));
    assertTrue(sp.getForwardingTargetFields(0, 2).contains(1));
    assertTrue(sp.getForwardingTargetFields(0, 3).contains(2));
    assertTrue(sp.getForwardingTargetFields(0, 0).contains(4));
    forwardedFields[0] = "f1.f0->f0; f0->f2";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, deepNestedTupleType, nestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 1).contains(0));
    assertTrue(sp.getForwardingTargetFields(0, 2).contains(1));
    assertTrue(sp.getForwardingTargetFields(0, 3).contains(2));
    assertTrue(sp.getForwardingTargetFields(0, 0).contains(4));
    forwardedFields[0] = "1.0->0; 0->2";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, deepNestedTupleType, nestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 1).contains(0));
    assertTrue(sp.getForwardingTargetFields(0, 2).contains(1));
    assertTrue(sp.getForwardingTargetFields(0, 3).contains(2));
    assertTrue(sp.getForwardingTargetFields(0, 0).contains(4));
    forwardedFields[0] = "f1.f0.f1; f1.f1; f2";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, deepNestedTupleType, deepNestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 2).contains(2));
    assertTrue(sp.getForwardingTargetFields(0, 4).contains(4));
    assertTrue(sp.getForwardingTargetFields(0, 6).contains(6));
    assertTrue(sp.getForwardingTargetFields(0, 0).size() == 0);
    assertTrue(sp.getForwardingTargetFields(0, 1).size() == 0);
    assertTrue(sp.getForwardingTargetFields(0, 3).size() == 0);
    assertTrue(sp.getForwardingTargetFields(0, 5).size() == 0);
    forwardedFields[0] = "f1.f0.*; f1.f2";
    sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, forwardedFields, null, null, deepNestedTupleType, deepNestedTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 1).contains(1));
    assertTrue(sp.getForwardingTargetFields(0, 2).contains(2));
    assertTrue(sp.getForwardingTargetFields(0, 3).contains(3));
    assertTrue(sp.getForwardingTargetFields(0, 5).contains(5));
    assertTrue(sp.getForwardingTargetFields(0, 0).size() == 0);
    assertTrue(sp.getForwardingTargetFields(0, 4).size() == 0);
    assertTrue(sp.getForwardingTargetFields(0, 6).size() == 0);
}
Also used : SingleInputSemanticProperties(org.apache.flink.api.common.operators.SingleInputSemanticProperties) Test(org.junit.Test)

Example 35 with SingleInputSemanticProperties

use of org.apache.flink.api.common.operators.SingleInputSemanticProperties in project flink by apache.

the class SemanticPropUtilTest method testNonForwardedSpaces.

@Test
public void testNonForwardedSpaces() {
    String[] nonForwardedFields = { " f1 ;   f2" };
    SingleInputSemanticProperties sp = new SingleInputSemanticProperties();
    SemanticPropUtil.getSemanticPropsSingleFromString(sp, null, nonForwardedFields, null, threeIntTupleType, threeIntTupleType);
    assertTrue(sp.getForwardingTargetFields(0, 0).contains(0));
    assertTrue(sp.getForwardingTargetFields(0, 1).size() == 0);
    assertTrue(sp.getForwardingTargetFields(0, 2).size() == 0);
}
Also used : SingleInputSemanticProperties(org.apache.flink.api.common.operators.SingleInputSemanticProperties) Test(org.junit.Test)

Aggregations

SingleInputSemanticProperties (org.apache.flink.api.common.operators.SingleInputSemanticProperties)149 Test (org.junit.Test)131 FieldSet (org.apache.flink.api.common.operators.util.FieldSet)57 FieldList (org.apache.flink.api.common.operators.util.FieldList)26 Ordering (org.apache.flink.api.common.operators.Ordering)15 Plan (org.apache.flink.api.common.Plan)14 ExecutionEnvironment (org.apache.flink.api.java.ExecutionEnvironment)14 MapOperatorBase (org.apache.flink.api.common.operators.base.MapOperatorBase)12 Tuple3 (org.apache.flink.api.java.tuple.Tuple3)12 DualInputSemanticProperties (org.apache.flink.api.common.operators.DualInputSemanticProperties)5 SemanticProperties (org.apache.flink.api.common.operators.SemanticProperties)5 Internal (org.apache.flink.annotation.Internal)4 SelectorFunctionKeys (org.apache.flink.api.common.operators.Keys.SelectorFunctionKeys)4 InvalidSemanticAnnotationException (org.apache.flink.api.common.operators.SemanticProperties.InvalidSemanticAnnotationException)4 Matcher (java.util.regex.Matcher)3 UnaryOperatorInformation (org.apache.flink.api.common.operators.UnaryOperatorInformation)3 FlatFieldDescriptor (org.apache.flink.api.common.typeutils.CompositeType.FlatFieldDescriptor)3 InvalidFieldReferenceException (org.apache.flink.api.common.typeutils.CompositeType.InvalidFieldReferenceException)3 Tuple2 (org.apache.flink.api.java.tuple.Tuple2)3 Annotation (java.lang.annotation.Annotation)2