def test_write_container_file(): for schema, value in SCHEMAS_TO_VALIDATE: fp = io.BytesIO() with avrolight.ContainerWriter(fp, json.loads(schema)) as writer: for _ in range(10): writer.write(value) values = list(avrolight.read_container(io.BytesIO(fp.getvalue()))) assert_that(values, has_length(10)) assert_that(values[0], equal_to(value))
def test_read_container_file(): import avro.io import avro.schema import avro.datafile for schema, value in SCHEMAS_TO_VALIDATE: writer = io.BytesIO() datum_writer = avro.io.DatumWriter(avro.schema.Parse(schema)) container = avro.datafile.DataFileWriter(writer, datum_writer, avro.schema.Parse(schema)) for _ in range(10): container.append(value) container.flush() values = list(avrolight.read_container(io.BytesIO(writer.getvalue()))) assert_that(values, has_length(10)) assert_that(values[0], equal_to(value))