def test_read_from_avro(self):
path = self._write_data()
with TestPipeline() as p:
assert_that(
p | avroio.ReadFromAvro(path, use_fastavro=self.use_fastavro),
equal_to(self.RECORDS))
def test_source_transform(self):
path = self._write_data()
with beam.Pipeline('DirectRunner') as p:
assert_that(p | avroio.ReadFromAvro(path), equal_to(self.RECORDS))
