test(pandas): memtable for pandas+dask backends supported

Author: anjakefala

ibis/backends/tests/test_array.py

@@ -473,11 +473,8 @@ def test_array_slice(con, start, stop):
 )
 @pytest.mark.notimpl(
     ["dask", "pandas"],
-    raises=com.UnboundExpressionError,
-    reason=(
-        "Node of type 'InMemoryTable' has no data bound to it. "
-        "You probably tried to execute an expression without a data source."
-    ),
+    raises=com.OperationNotDefinedError,
+    reason="Operation 'ArrayMap' is not implemented for this backend",
 )
 @pytest.mark.notimpl(
     ["sqlite", "mysql"],
@@ -508,11 +505,8 @@ def test_array_map(backend, con):
 )
 @pytest.mark.notimpl(
     ["dask", "pandas"],
-    raises=com.UnboundExpressionError,
-    reason=(
-        "Node of type 'InMemoryTable' has no data bound to it. "
-        "You probably tried to execute an expression without a data source."
-    ),
+    raises=com.OperationNotDefinedError,
+    reason="Operation 'ArrayMap' is not implemented for this backend'",
 )
 @pytest.mark.notimpl(
     ["sqlite", "mysql"],

ibis/backends/tests/test_client.py

@@ -719,13 +719,13 @@ def test_deprecated_path_argument(backend_name, tmp_path):
         ),
     ],
 )
-@pytest.mark.notimpl(["dask", "datafusion", "pandas"])
+@pytest.mark.notimpl(["datafusion"])
 def test_in_memory_table(backend, con, expr, expected):
     result = con.execute(expr)
     backend.assert_frame_equal(result, expected)
 
 
-@pytest.mark.notimpl(["dask", "datafusion", "pandas"])
+@pytest.mark.notimpl(["datafusion"])
 def test_filter_memory_table(backend, con):
     t = ibis.memtable([(1, 2), (3, 4), (5, 6)], columns=["x", "y"])
     expr = t.filter(t.x > 1)
@@ -734,7 +734,7 @@ def test_filter_memory_table(backend, con):
     backend.assert_frame_equal(result, expected)
 
 
-@pytest.mark.notimpl(["dask", "datafusion", "pandas"])
+@pytest.mark.notimpl(["datafusion"])
 def test_agg_memory_table(con):
     t = ibis.memtable([(1, 2), (3, 4), (5, 6)], columns=["x", "y"])
     expr = t.x.count()

ibis/backends/tests/test_export.py

@@ -151,15 +151,15 @@ def test_to_pyarrow_batches_borked_types(batting):
     assert len(batch) == 42
 
 
-@pytest.mark.notimpl(["dask", "datafusion", "impala", "pandas", "pyspark"])
+@pytest.mark.notimpl(["dask", "datafusion", "impala", "pyspark"])
 def test_to_pyarrow_memtable(con):
     expr = ibis.memtable({"x": [1, 2, 3]})
     table = con.to_pyarrow(expr)
     assert isinstance(table, pa.Table)
     assert len(table) == 3
 
 
-@pytest.mark.notimpl(["dask", "datafusion", "impala", "pandas", "pyspark"])
+@pytest.mark.notimpl(["dask", "datafusion", "impala", "pyspark"])
 def test_to_pyarrow_batches_memtable(con):
     expr = ibis.memtable({"x": [1, 2, 3]})
     n = 0

ibis/backends/tests/test_generic.py

@@ -916,7 +916,6 @@ def test_literal_na(con, dtype):
 
 
 @pytest.mark.notimpl(["datafusion"], raises=com.OperationNotDefinedError)
-@pytest.mark.notimpl(["dask"], raises=com.UnboundExpressionError)
 def test_memtable_bool_column(backend, con, monkeypatch):
     monkeypatch.setattr(ibis.options, "default_backend", con)
 
@@ -925,7 +924,6 @@ def test_memtable_bool_column(backend, con, monkeypatch):
 
 
 @pytest.mark.notimpl(["datafusion"], raises=com.OperationNotDefinedError)
-@pytest.mark.notimpl(["dask"], raises=com.UnboundExpressionError)
 @pytest.mark.broken(
     ["druid"],
     raises=AssertionError,

ibis/backends/tests/test_join.py

@@ -199,7 +199,7 @@ def test_semi_join_topk(batting, awards_players):
     assert not expr.limit(5).execute().empty
 
 
-@pytest.mark.notimpl(["dask", "datafusion", "druid", "pandas"])
+@pytest.mark.notimpl(["dask", "datafusion", "druid"])
 @pytest.mark.broken(
     ["duckdb"],
     raises=exc.IbisTypeError,
@@ -214,7 +214,7 @@ def test_join_with_pandas(batting, awards_players):
     assert df.yearID.nunique() == 7
 
 
-@pytest.mark.notimpl(["dask", "datafusion", "pandas"])
+@pytest.mark.notimpl(["dask", "datafusion"])
 def test_join_with_pandas_non_null_typed_columns(batting, awards_players):
     batting_filt = batting[lambda t: t.yearID < 1900][["yearID"]]
     awards_players_filt = awards_players[lambda t: t.yearID < 1900][