Function shifted_sum

dask/dataframe/tests/test_rolling.py:51–54

(df, before, after, c=0)

def shifted_sum(df, before, after, c=0):
    a = df.shift(before)
    b = df.shift(-after)
    return df + a + b + c


@pytest.mark.parametrize("npartitions", [1, 4])