fix test failures

Author: sycai

tests/system/test_gbq.py

@@ -16,7 +16,7 @@
 import pytest
 import pytz
 
-from pandas_gbq import gbq
+from pandas_gbq import exceptions, gbq
 import pandas_gbq.schema
 
 TABLE_ID = "new_test"
@@ -238,7 +238,7 @@ def test_read_gbq_raises_invalid_index_column(self, project_id):
             )
 
     def test_malformed_query(self, project_id):
-        with pytest.raises(gbq.GenericGBQException):
+        with pytest.raises(exceptions.GenericGBQException):
             gbq.read_gbq(
                 "SELCET * FORM [publicdata:samples.shakespeare]",
                 project_id=project_id,
@@ -247,7 +247,7 @@ def test_malformed_query(self, project_id):
             )
 
     def test_bad_project_id(self):
-        with pytest.raises(gbq.GenericGBQException):
+        with pytest.raises(exceptions.GenericGBQException):
             gbq.read_gbq(
                 "SELCET * FROM [publicdata:samples.shakespeare]",
                 project_id="not-my-project",
@@ -256,7 +256,7 @@ def test_bad_project_id(self):
             )
 
     def test_bad_table_name(self, project_id):
-        with pytest.raises(gbq.GenericGBQException):
+        with pytest.raises(exceptions.GenericGBQException):
             gbq.read_gbq(
                 "SELECT * FROM [publicdata:samples.nope]",
                 project_id=project_id,
@@ -316,7 +316,7 @@ def test_legacy_sql(self, project_id):
 
         # Test that a legacy sql statement fails when
         # setting dialect='standard'
-        with pytest.raises(gbq.GenericGBQException):
+        with pytest.raises(exceptions.GenericGBQException):
             gbq.read_gbq(
                 legacy_sql,
                 project_id=project_id,
@@ -341,7 +341,7 @@ def test_standard_sql(self, project_id):
 
         # Test that a standard sql statement fails when using
         # the legacy SQL dialect.
-        with pytest.raises(gbq.GenericGBQException):
+        with pytest.raises(exceptions.GenericGBQException):
             gbq.read_gbq(
                 standard_sql,
                 project_id=project_id,
@@ -1029,7 +1029,7 @@ def test_upload_data_with_invalid_user_schema_raises_error(self, project_id):
             {"name": "D", "type": "FLOAT"},
         ]
         destination_table = self.destination_table + test_id
-        with pytest.raises(gbq.GenericGBQException):
+        with pytest.raises(exceptions.GenericGBQException):
             gbq.to_gbq(
                 df,
                 destination_table,
@@ -1047,7 +1047,7 @@ def test_upload_data_with_missing_schema_fields_raises_error(self, project_id):
             {"name": "C", "type": "FLOAT"},
         ]
         destination_table = self.destination_table + test_id
-        with pytest.raises(gbq.GenericGBQException):
+        with pytest.raises(exceptions.GenericGBQException):
             gbq.to_gbq(
                 df,
                 destination_table,