chore: fix druid exception and test case

ibis/backends/druid/__init__.py

@@ -10,7 +10,6 @@
 
 import pydruid.db
 import sqlglot as sg
-from pydruid.db.exceptions import ProgrammingError
 
 import ibis.common.exceptions as com
 import ibis.expr.datatypes as dt
@@ -20,6 +19,7 @@
 from ibis.backends.sql.compilers import DruidCompiler
 from ibis.backends.sql.compilers.base import STAR
 from ibis.backends.sql.datatypes import DruidType
+from ibis.backends.tests.errors import PyDruidProgrammingError
 
 if TYPE_CHECKING:
     from collections.abc import Iterable, Mapping
@@ -139,7 +139,7 @@ def get_schema(
                 .from_(sg.table(table_name, db=database, catalog=catalog))
                 .sql(self.dialect)
             )
-        except ProgrammingError as e:
+        except PyDruidProgrammingError as e:
             if re.search(r"\bINVALID_INPUT\b", str(e)):
                 raise com.TableNotFound(table_name) from e
 

ibis/backends/tests/test_join.py

@@ -11,7 +11,6 @@
 import ibis
 import ibis.common.exceptions as com
 import ibis.expr.schema as sch
-from ibis.backends.tests.errors import PyDruidProgrammingError
 
 sqlite_right_or_full_mark = pytest.mark.notyet(
     ["sqlite"],
@@ -292,7 +291,7 @@ def test_join_with_trivial_predicate(awards_players, predicate, how, pandas_valu
     assert len(result) == len(expected)
 
 
-@pytest.mark.notimpl(["druid"], raises=PyDruidProgrammingError)
+@pytest.mark.notimpl(["druid"], raises=com.TableNotFound)
 @pytest.mark.parametrize(
     ("how", "nrows", "gen_right", "keys"),
     [