| 108 | |
| 109 | |
| 110 | def test_resample_agg(df, pdf): |
| 111 | def my_sum(vals, foo=None, *, bar=None): |
| 112 | return vals.sum() |
| 113 | |
| 114 | result = df.resample("2min").agg(my_sum, "foo", bar="bar") |
| 115 | expected = pdf.resample("2min").agg(my_sum, "foo", bar="bar") |
| 116 | assert_eq(result, expected) |
| 117 | |
| 118 | result = df.resample("2min").agg(my_sum)["foo"] |
| 119 | expected = pdf.resample("2min").agg(my_sum)["foo"] |
| 120 | assert_eq(result, expected) |
| 121 | |
| 122 | # simplify up disabled for `agg`, function may access other columns |
| 123 | q = df.resample("2min").agg(my_sum)["foo"].simplify() |
| 124 | eq = df["foo"].resample("2min").agg(my_sum).simplify() |
| 125 | assert q._name != eq._name |
| 126 | |
| 127 | |
| 128 | @pytest.mark.parametrize("method", ["count", "nunique", "size", "sum"]) |