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_shift(): df = DataFrame({"A": ["a", "b", "c"], "B": [1, 3, 5]}) a = df.shift(1) b = DataFrame({"A": [None, "a", "b"], "B": [None, 1, 3]}, nullable=True) assert a.frame_equal(b, null_equal=True)