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))
