use of org.apache.flink.api.common.operators.util.FieldSet in project flink by apache.
the class RequestedLocalPropertiesFilteringTest method testDualGroupingPreserved.
@Test
public void testDualGroupingPreserved() {
DualInputSemanticProperties dprops = new DualInputSemanticProperties();
SemanticPropUtil.getSemanticPropsDualFromString(dprops, new String[] { "1->0;3;2->4" }, new String[] { "0->7;1" }, null, null, null, null, tupleInfo, tupleInfo, tupleInfo);
RequestedLocalProperties lprops1 = new RequestedLocalProperties();
lprops1.setGroupedFields(new FieldSet(0, 3, 4));
RequestedLocalProperties lprops2 = new RequestedLocalProperties();
lprops2.setGroupedFields(new FieldSet(7, 1));
RequestedLocalProperties filtered1 = lprops1.filterBySemanticProperties(dprops, 0);
RequestedLocalProperties filtered2 = lprops2.filterBySemanticProperties(dprops, 1);
assertNotNull(filtered1);
assertNotNull(filtered1.getGroupedFields());
assertEquals(3, filtered1.getGroupedFields().size());
assertTrue(filtered1.getGroupedFields().contains(1));
assertTrue(filtered1.getGroupedFields().contains(2));
assertTrue(filtered1.getGroupedFields().contains(3));
assertNull(filtered1.getOrdering());
assertNotNull(filtered2);
assertNotNull(filtered2.getGroupedFields());
assertEquals(2, filtered2.getGroupedFields().size());
assertTrue(filtered2.getGroupedFields().contains(0));
assertTrue(filtered2.getGroupedFields().contains(1));
assertNull(filtered2.getOrdering());
}
use of org.apache.flink.api.common.operators.util.FieldSet in project flink by apache.
the class RequestedLocalPropertiesFilteringTest method testGroupingErased.
@Test
public void testGroupingErased() {
SingleInputSemanticProperties sProps = new SingleInputSemanticProperties();
SemanticPropUtil.getSemanticPropsSingleFromString(sProps, new String[] { "0;2" }, null, null, tupleInfo, tupleInfo);
RequestedLocalProperties rlProp = new RequestedLocalProperties();
rlProp.setGroupedFields(new FieldSet(0, 2, 3));
RequestedLocalProperties filtered = rlProp.filterBySemanticProperties(sProps, 0);
assertNull(filtered);
}
use of org.apache.flink.api.common.operators.util.FieldSet in project flink by apache.
the class RequestedLocalPropertiesFilteringTest method testGroupingPreserved1.
@Test
public void testGroupingPreserved1() {
SingleInputSemanticProperties sProps = new SingleInputSemanticProperties();
SemanticPropUtil.getSemanticPropsSingleFromString(sProps, new String[] { "0;2;3" }, null, null, tupleInfo, tupleInfo);
RequestedLocalProperties rlProp = new RequestedLocalProperties();
rlProp.setGroupedFields(new FieldSet(0, 2, 3));
RequestedLocalProperties filtered = rlProp.filterBySemanticProperties(sProps, 0);
assertNotNull(filtered);
assertNotNull(filtered.getGroupedFields());
assertEquals(3, filtered.getGroupedFields().size());
assertTrue(filtered.getGroupedFields().contains(0));
assertTrue(filtered.getGroupedFields().contains(2));
assertTrue(filtered.getGroupedFields().contains(3));
assertNull(filtered.getOrdering());
}
use of org.apache.flink.api.common.operators.util.FieldSet in project flink by apache.
the class RequestedLocalPropertiesFilteringTest method testGroupingPreserved2.
@Test
public void testGroupingPreserved2() {
SingleInputSemanticProperties sProps = new SingleInputSemanticProperties();
SemanticPropUtil.getSemanticPropsSingleFromString(sProps, new String[] { "3->0;5->2;1->3" }, null, null, tupleInfo, tupleInfo);
RequestedLocalProperties rlProp = new RequestedLocalProperties();
rlProp.setGroupedFields(new FieldSet(0, 2, 3));
RequestedLocalProperties filtered = rlProp.filterBySemanticProperties(sProps, 0);
assertNotNull(filtered);
assertNotNull(filtered.getGroupedFields());
assertEquals(3, filtered.getGroupedFields().size());
assertTrue(filtered.getGroupedFields().contains(3));
assertTrue(filtered.getGroupedFields().contains(5));
assertTrue(filtered.getGroupedFields().contains(1));
assertNull(filtered.getOrdering());
}
use of org.apache.flink.api.common.operators.util.FieldSet in project flink by apache.
the class RequestedLocalPropertiesFilteringTest method testInvalidInputIndex.
@Test(expected = IndexOutOfBoundsException.class)
public void testInvalidInputIndex() {
SingleInputSemanticProperties sProps = new SingleInputSemanticProperties();
SemanticPropUtil.getSemanticPropsSingleFromString(sProps, new String[] { "1; 4" }, null, null, tupleInfo, tupleInfo);
RequestedLocalProperties rlProp = new RequestedLocalProperties();
rlProp.setGroupedFields(new FieldSet(1, 4));
rlProp.filterBySemanticProperties(sProps, 1);
}
Aggregations