Fix tests

Author: mroeschke

pandas/tests/extension/test_arrow.py

@@ -2326,7 +2326,8 @@ def test_str_split():
         {
             0: ArrowExtensionArray(pa.array(["a", "a", None])),
             1: ArrowExtensionArray(pa.array(["cbcb", "cbcb", None])),
-        }
+        },
+        columns=range(2),
     )
     tm.assert_frame_equal(result, expected)
 
@@ -2335,7 +2336,8 @@ def test_str_split():
         {
             0: ArrowExtensionArray(pa.array(["a", "a2cbcb", None])),
             1: ArrowExtensionArray(pa.array(["cbcb", None, None])),
-        }
+        },
+        columns=range(2),
     )
     tm.assert_frame_equal(result, expected)
 
@@ -2360,7 +2362,8 @@ def test_str_rsplit():
         {
             0: ArrowExtensionArray(pa.array(["a1cb", "a2cb", None])),
             1: ArrowExtensionArray(pa.array(["b", "b", None])),
-        }
+        },
+        columns=range(2),
     )
     tm.assert_frame_equal(result, expected)
 
@@ -2369,7 +2372,8 @@ def test_str_rsplit():
         {
             0: ArrowExtensionArray(pa.array(["a", "a2cbcb", None])),
             1: ArrowExtensionArray(pa.array(["cbcb", None, None])),
-        }
+        },
+        columns=range(2),
     )
     tm.assert_frame_equal(result, expected)
 

pandas/tests/groupby/test_groupby.py

@@ -517,7 +517,11 @@ def test_as_index_select_column():
         lambda x: x.cumsum()
     )
     expected = Series(
-        [2, 6, 6], name="B", index=MultiIndex.from_arrays([[0, 0, 1], RangeIndex(3)])
+        [2, 6, 6],
+        name="B",
+        index=MultiIndex(
+            levels=[RangeIndex(2), RangeIndex(3)], codes=[[0, 0, 1], [0, 1, 2]]
+        ),
     )
     tm.assert_series_equal(result, expected)