コード例 #1
0
 @Test
 public void testFormatShakespeareOutputFn() {
   DoFnTester<KV<String, String>, TableRow> formatShakespeareOutputFn =
       DoFnTester.of(new FormatShakespeareOutputFn());
   List<TableRow> results = formatShakespeareOutputFn.processBatch(COMBINED_TUPLES_ARRAY);
   Assert.assertThat(results, CoreMatchers.hasItem(resultRow1));
   Assert.assertThat(results, CoreMatchers.hasItem(resultRow2));
 }
コード例 #2
0
 @Test
 public void testExtractLargeWordsFn() {
   DoFnTester<TableRow, KV<String, String>> extractLargeWordsFn =
       DoFnTester.of(new ExtractLargeWordsFn());
   List<KV<String, String>> results = extractLargeWordsFn.processBatch(ROWS_ARRAY);
   Assert.assertThat(results, CoreMatchers.hasItem(tuple1));
   Assert.assertThat(results, CoreMatchers.hasItem(tuple2));
   Assert.assertThat(results, CoreMatchers.hasItem(tuple3));
 }