updates to enum import

Author: chalmerlowe

tests/unit/job/test_query_pandas.py

@@ -22,6 +22,7 @@
 from ..helpers import make_connection
 from .helpers import _make_client
 from .helpers import _make_job_resource
+from google.cloud.bigquery.enums import DefaultPandasDTypes
 
 try:
     from google.cloud import bigquery_storage
@@ -30,6 +31,7 @@
 except (ImportError, AttributeError):
     bigquery_storage = None
 
+
 try:
     import shapely
 except (ImportError, AttributeError):
@@ -1000,8 +1002,6 @@ def test_query_job_to_geodataframe_delegation(wait_for_query):
     max_results = 42
     geography_column = "g"
 
-    from google.cloud.bigquery.enums import DefaultPandasDTypes
-
     df = job.to_geodataframe(
         bqstorage_client=bqstorage_client,
         dtypes=dtypes,

tests/unit/test_table.py

@@ -31,6 +31,7 @@
 from google.cloud.bigquery import exceptions
 from google.cloud.bigquery import external_config
 from google.cloud.bigquery import schema
+from google.cloud.bigquery.enums import DefaultPandasDTypes
 from google.cloud.bigquery.table import TableReference
 from google.cloud.bigquery.dataset import DatasetReference
 
@@ -5633,8 +5634,6 @@ def test_rowiterator_to_geodataframe_delegation(self, to_dataframe):
             geography_column=geography_column,
         )
 
-        from google.cloud.bigquery.enums import DefaultPandasDTypes
-
         to_dataframe.assert_called_once_with(
             bqstorage_client,
             dtypes,