test(polars): add test for cleaning up the SQL context when a table i…

…s dropped

ibis/backends/polars/tests/test_client.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+import pytest
+
+from ibis.backends.tests.errors import PolarsSQLInterfaceError
+from ibis.util import gen_name
+
+
+def test_cannot_run_sql_after_drop(con):
+    t = con.table("functional_alltypes")
+    n = t.count().execute()
+
+    name = gen_name("polars_dot_sql")
+    con.create_table(name, t)
+
+    sql = f"SELECT COUNT(*) FROM {name}"
+
+    expr = con.sql(sql)
+    result = expr.execute()
+    assert result.iat[0, 0] == n
+
+    con.drop_table(name)
+    with pytest.raises(PolarsSQLInterfaceError):
+        con.sql(sql)

ibis/backends/tests/errors.py

@@ -65,10 +65,11 @@
     from polars.exceptions import InvalidOperationError as PolarsInvalidOperationError
     from polars.exceptions import PanicException as PolarsPanicException
     from polars.exceptions import SchemaError as PolarsSchemaError
+    from polars.exceptions import SQLInterfaceError as PolarsSQLInterfaceError
 except ImportError:
     PolarsComputeError = PolarsPanicException = PolarsInvalidOperationError = (
         PolarsSchemaError
-    ) = PolarsColumnNotFoundError = None
+    ) = PolarsColumnNotFoundError = PolarsSQLInterfaceError = None
 
 try:
     from pyarrow import ArrowInvalid, ArrowNotImplementedError