Search in sources :

Example 1 with FilterSingleMonthDataFn

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

the class FilterExamplesTest method testFilterSingleMonthDataFn.

@Test
public void testFilterSingleMonthDataFn() throws Exception {
    DoFnTester<TableRow, TableRow> filterSingleMonthDataFn = DoFnTester.of(new FilterSingleMonthDataFn(7));
    List<TableRow> results = filterSingleMonthDataFn.processBundle(PROJROWS_ARRAY);
    Assert.assertThat(results, CoreMatchers.hasItem(outRow2));
}
Also used : TableRow(com.google.api.services.bigquery.model.TableRow) FilterSingleMonthDataFn(org.apache.beam.examples.cookbook.FilterExamples.FilterSingleMonthDataFn) Test(org.junit.Test)

Aggregations

TableRow (com.google.api.services.bigquery.model.TableRow)1 FilterSingleMonthDataFn (org.apache.beam.examples.cookbook.FilterExamples.FilterSingleMonthDataFn)1 Test (org.junit.Test)1