def test_to_pandas():
df = get_complete_df()
df.to_arrow()
df.to_pandas()
# test shifted df
df.shift(2).to_pandas()
df = DataFrame({"col": Series([True, False, True])})
df.shift(2).to_pandas()
def test_df_stats():
df = get_complete_df()
df.var()
df.std()
df.min()
df.max()
df.sum()
df.mean()
df.median()
df.quantile(0.4)
def test_to_from_buffer():
df = get_complete_df()
df = df.drop("strings_nulls")
for to_fn, from_fn in zip([df.to_parquet, df.to_csv],
[df.read_parquet, df.read_csv]):
f = io.BytesIO()
to_fn(f)
f.seek(0)
df_1 = from_fn(f)
assert df.frame_equal(df_1, null_equal=True)
def test_to_html():
df = get_complete_df()
# check if it does not panic/ error
df._repr_html_()
def test_argsort_by():
df = get_complete_df()
a = df[pl.argsort_by(["int_nulls", "floats"], reverse=[False, True])]["int_nulls"]
assert a == [1, 0, 3]
