test(backends): fix pandas and dask tests

ibis-project · Aug 26, 2024 · 7393c31 · 7393c31
1 parent 6e2c429
commit 7393c31
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/ibis/backends/dask/tests/conftest.py b/ibis/backends/dask/tests/conftest.py
@@ -376,7 +376,7 @@ def batting(lahman):
 def sel_cols(batting):
     cols = batting.columns
     start, end = cols.index("AB"), cols.index("H") + 1
-    return ["playerID", "yearID", "teamID", "G"] + cols[start:end]
+    return ["playerID", "yearID", "teamID", "G", *cols[start:end]]
 
 
 @pytest.fixture(scope="module")

diff --git a/ibis/backends/dask/tests/test_arrays.py b/ibis/backends/dask/tests/test_arrays.py
@@ -70,7 +70,7 @@ def test_array_collect_rolling_partitioned(t, df):
             }
         ),
         npartitions=1,
-    )[expr.columns]
+    )[list(expr.columns)]
     tm.assert_frame_equal(
         result.compute().reset_index(drop=True),
         expected.compute().reset_index(drop=True),

diff --git a/ibis/backends/pandas/tests/conftest.py b/ibis/backends/pandas/tests/conftest.py
@@ -301,7 +301,7 @@ def batting(lahman):
 def sel_cols(batting):
     cols = batting.columns
     start, end = cols.index("AB"), cols.index("H") + 1
-    return ["playerID", "yearID", "teamID", "G"] + cols[start:end]
+    return ["playerID", "yearID", "teamID", "G", *cols[start:end]]
 
 
 @pytest.fixture(scope="module")

diff --git a/ibis/backends/pandas/tests/test_arrays.py b/ibis/backends/pandas/tests/test_arrays.py
@@ -82,7 +82,7 @@ def test_array_collect_rolling_partitioned(t, df):
             "plain_int64": [1, 2, 3],
             "collected": [[4.0], [4.0, 5.0], [5.0, 6.0]],
         }
-    )[expr.columns]
+    )[list(expr.columns)]
     tm.assert_frame_equal(result, expected)