Update test_regression_whitelist_methods, remove specialized test

Author: alexcwatt

pandas/tests/groupby/test_function.py

@@ -490,25 +490,6 @@ def test_ops_general():
             raise
 
 
-@pytest.mark.parametrize('as_index', [True, False])
-def test_groupby_std(as_index):
-    # GH 10355: Test that std does not affect the groupby column
-    df = pd.DataFrame({
-        'a': [1, 1, 1, 2, 2, 2, 3, 3, 3],
-        'b': [1, 2, 3, 3, 5, 7, 7, 8, 9],
-    })
-    result = df.groupby('a', as_index=as_index).std()
-    expected = pd.DataFrame({
-        'a': [1, 2, 3],
-        'b': [1, 2, 1]
-    })
-
-    if as_index:
-        expected = expected.set_index('a')
-
-    tm.assert_frame_equal(result, expected)
-
-
 def test_max_nan_bug():
     raw = """,Date,app,File
 -04-23,2013-04-23 00:00:00,,log080001.log

pandas/tests/groupby/test_whitelist.py

@@ -180,7 +180,7 @@ def test_regression_whitelist_methods(
     else:
         frame = raw_frame.T
 
-    groupby_kwargs = {'level': level, 'axis': axis, 'sort': sort} #, 'as_index': as_index}
+    groupby_kwargs = {'level': level, 'axis': axis, 'sort': sort, 'as_index': as_index}
     group_op_kwargs = {}
     frame_op_kwargs = {'level': level, 'axis': axis}
     if op in AGG_FUNCTIONS_WITH_SKIPNA:
@@ -190,12 +190,17 @@ def test_regression_whitelist_methods(
     grouped = frame.groupby(**groupby_kwargs)
     result = getattr(grouped, op)(**group_op_kwargs)
     expected = getattr(frame, op)(**frame_op_kwargs)
-    if as_index:
-        pass
 
     if sort:
         expected = expected.sort_index(axis=axis, level=level)
 
+    if not as_index:
+        expected = expected.reset_index()
+        if level == 0:
+            expected = expected.drop(columns=['first'])
+        if level == 1:
+            expected = expected.drop(columns=['second'])
+
     tm.assert_frame_equal(result, expected)