@Test public void testBasicRecordWriter() { DatasetRecordWriter<GenericData.Record> recordWriter; recordWriter = new DatasetRecordWriter<GenericData.Record>(dataset, false); ImmutableList<Integer> counts = ImmutableList.of(1, 2, 3, 4, 5, 6, 7, 8, 9, 10); for (Integer count : counts) { GenericData.Record record = new GenericData.Record(STATS_SCHEMA); record.put("count", count); record.put("name", "name"+count); recordWriter.write(record, null); } recordWriter.close(null); }