From b54f5281ad6604e348b839929eb703ba84f20742 Mon Sep 17 00:00:00 2001
From: Phillip Cloud <417981+cpcloud@users.noreply.github.com>
Date: Mon, 16 Sep 2024 07:21:17 -0400
Subject: [PATCH] test(bigquery): check the correct exception for missing
 tables (#10137)

---
 ibis/backends/bigquery/tests/system/test_connect.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ibis/backends/bigquery/tests/system/test_connect.py b/ibis/backends/bigquery/tests/system/test_connect.py
index 4e6cc5c19d37..f62d416ba707 100644
--- a/ibis/backends/bigquery/tests/system/test_connect.py
+++ b/ibis/backends/bigquery/tests/system/test_connect.py
@@ -11,6 +11,7 @@
 from google.cloud import bigquery_storage_v1 as bqstorage
 
 import ibis
+import ibis.common.exceptions as exc
 
 
 def test_repeated_project_name(project_id, credentials):
@@ -230,7 +231,7 @@ def test_client_with_regional_endpoints(project_id, credentials, dataset_id):
     )
 
     # Fails because dataset not in Tokyo.
-    with pytest.raises(gexc.NotFound, match=dataset_id):
+    with pytest.raises(exc.TableNotFound, match=dataset_id):
         tokyo_con.table(f"{dataset_id}.functional_alltypes")
 
     # Succeeds because dataset is in Tokyo.