(op, expected)
| 73 | ], |
| 74 | ) |
| 75 | def test_groupby_reductions(op, expected): |
| 76 | df = DataFrame( |
| 77 | { |
| 78 | "A": ["a", "b", "b"], |
| 79 | "B": array([1, None, 3], dtype="Int64"), |
| 80 | } |
| 81 | ) |
| 82 | result = getattr(df.groupby("A"), op)() |
| 83 | expected = DataFrame(expected, index=pd.Index(["a", "b"], name="A"), columns=["B"]) |
| 84 | |
| 85 | tm.assert_frame_equal(result, expected) |
| 86 | |
| 87 | |
| 88 | @pytest.mark.parametrize( |