Skip to content
Prev Previous commit
Next Next commit
Run black
  • Loading branch information
mproszewska committed May 5, 2020
commit e2b0f56728ebfa056cb79b8b53192f583e7731d5
3 changes: 2 additions & 1 deletion pandas/tests/window/test_apply.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,11 +139,12 @@ def test_invalid_kwargs_nopython():
lambda x: x, kwargs={"a": 1}, engine="numba", raw=True
)


def test_apply_kwargs():
def foo(x, par):
return np.sum(x + par)

df = DataFrame({"gr": [1, 1], "a": [1, 2]})
result = df.groupby("gr")["a"].rolling(1).apply(foo, kwargs={"par": 10})
expected = np.array([11.,12.])
expected = np.array([11.0, 12.0])
np.testing.assert_array_equal(result.values, expected)