@Test
public void createRecordReader() {

  assertThat(
          input.createRecordReader(new DirectBigQueryInputSplit("foo", "schema", 7), taskContext))
      .isInstanceOf(DirectBigQueryRecordReader.class);
}