Examples with ProjectionFn org.apache.beam.examples.cookbook.FilterExamples.ProjectionFn used on opensource projects

Example 1 with ProjectionFn

use of org.apache.beam.examples.cookbook.FilterExamples.ProjectionFn in project beam by apache.

the class FilterExamplesTest method testProjectionFn.

@Test
public void testProjectionFn() throws Exception {
    DoFnTester<TableRow, TableRow> projectionFn = DoFnTester.of(new ProjectionFn());
    List<TableRow> results = projectionFn.processBundle(ROWS_ARRAY);
    Assert.assertThat(results, CoreMatchers.hasItem(outRow1));
    Assert.assertThat(results, CoreMatchers.hasItem(outRow2));
    Assert.assertThat(results, CoreMatchers.hasItem(outRow3));
}