def test_from_to_file(io_test_dir: str, example_df: pl.DataFrame, compressions: List[str]) -> None: f = os.path.join(io_test_dir, "small.avro") for compression in compressions: example_df.write_avro(f, compression=compression) # type: ignore df_read = pl.read_avro(str(f)) assert example_df.frame_equal(df_read)
def test_from_to_buffer(example_df: pl.DataFrame, compressions: List[str]) -> None: for compression in compressions: buf = io.BytesIO() example_df.write_avro(buf, compression=compression) # type: ignore buf.seek(0) read_df = pl.read_avro(buf) assert example_df.frame_equal(read_df)