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]