Fix tests

Author: mroeschke

pandas/tests/groupby/test_groupby.py

@@ -116,8 +116,9 @@ def f(x, q=None, axis=0):
     expected_seq = df_grouped.quantile([0.4, 0.8])
     if not as_index:
         # apply treats the op as a transform; .quantile knows it's a reduction
-        apply_result = apply_result.reset_index()
-        apply_result["level_0"] = [1, 1, 2, 2]
+        apply_result.index = range(4)
+        apply_result.insert(loc=0, column="level_0", value=[1, 1, 2, 2])
+        apply_result.insert(loc=1, column="level_1", value=[0.4, 0.8, 0.4, 0.8])
     tm.assert_frame_equal(apply_result, expected_seq, check_names=False)
 
     agg_result = df_grouped.agg(f, q=80)
@@ -522,13 +523,7 @@ def test_as_index_select_column():
     result = df.groupby("A", as_index=False, group_keys=True)["B"].apply(
         lambda x: x.cumsum()
     )
-    expected = Series(
-        [2, 6, 6],
-        name="B",
-        index=MultiIndex(
-            levels=[RangeIndex(2), RangeIndex(3)], codes=[[0, 0, 1], [0, 1, 2]]
-        ),
-    )
+    expected = Series([2, 6, 6], name="B", index=range(3))
     tm.assert_series_equal(result, expected)