def test_to_from_buffer(df: pl.DataFrame) -> None:
for buf in (io.BytesIO(), io.StringIO()):
df.write_json(buf)
buf.seek(0)
read_df = pl.read_json(buf)
read_df = read_df.with_columns(
[pl.col("cat").cast(pl.Categorical),
pl.col("time").cast(pl.Time)])
assert_frame_equal_local_categoricals(df, read_df)
def test_write_json2(df: pl.DataFrame) -> None:
# text-based conversion loses time info
df = df.select(pl.all().exclude(["cat", "time"]))
s = df.write_json(to_string=True)
f = io.BytesIO()
f.write(s.encode())
f.seek(0)
out = pl.read_json(f)
assert df.frame_equal(out, null_equal=True)
file = io.BytesIO()
df.write_json(file)
file.seek(0)
out = pl.read_json(file)
assert df.frame_equal(out, null_equal=True)
def test_to_from_file(io_test_dir: str, df: pl.DataFrame) -> None:
f = os.path.join(io_test_dir, "small.json")
df.write_json(f)
out = pl.read_json(f)
assert out.frame_equal(df)
