diff --git a/pandas/core/groupby.py b/pandas/core/groupby.py index e8a9d6e49a066..182f75e53ca5d 100644 --- a/pandas/core/groupby.py +++ b/pandas/core/groupby.py @@ -65,6 +65,7 @@ 'mad', 'any', 'all', 'irow', 'take', + 'idxmax', 'idxmin', 'shift', 'tshift', 'ffill', 'bfill', 'pct_change', 'skew', diff --git a/pandas/tests/test_groupby.py b/pandas/tests/test_groupby.py index 942efdfc23740..7e54aa4e0813f 100644 --- a/pandas/tests/test_groupby.py +++ b/pandas/tests/test_groupby.py @@ -3264,6 +3264,7 @@ def test_groupby_whitelist(self): 'mad', 'any', 'all', 'irow', 'take', + 'idxmax', 'idxmin', 'shift', 'tshift', 'ffill', 'bfill', 'pct_change', 'skew', @@ -3284,6 +3285,7 @@ def test_groupby_whitelist(self): 'mad', 'any', 'all', 'irow', 'take', + 'idxmax', 'idxmin', 'shift', 'tshift', 'ffill', 'bfill', 'pct_change', 'skew', @@ -3413,7 +3415,7 @@ def test_tab_completion(self): 'resample', 'cummin', 'fillna', 'cumsum', 'cumcount', 'all', 'shift', 'skew', 'bfill', 'irow', 'ffill', 'take', 'tshift', 'pct_change', 'any', 'mad', 'corr', 'corrwith', - 'cov', 'dtypes', 'diff', + 'cov', 'dtypes', 'diff', 'idxmax', 'idxmin' ]) self.assertEqual(results, expected)