diff --git a/packages/google-cloud-bigquery/tests/unit/job/test_async_job_retry.py b/packages/google-cloud-bigquery/tests/unit/job/test_async_job_retry.py index 4d9cccf932c3..f317651c5f45 100644 --- a/packages/google-cloud-bigquery/tests/unit/job/test_async_job_retry.py +++ b/packages/google-cloud-bigquery/tests/unit/job/test_async_job_retry.py @@ -15,11 +15,10 @@ from unittest import mock import google.api_core.retry +import google.cloud.bigquery.job from google.api_core import exceptions from . import helpers -import google.cloud.bigquery.job - PROJECT = "test-project" JOB_ID = "test-job-id" @@ -110,7 +109,7 @@ def test_result_w_retry_wo_state(global_time_lock): predicate=custom_predicate, initial=0.001, maximum=0.001, - deadline=0.1, + deadline=1.0, ) assert job.result(retry=custom_retry) is job diff --git a/packages/google-cloud-bigquery/tests/unit/job/test_query_job_retry.py b/packages/google-cloud-bigquery/tests/unit/job/test_query_job_retry.py index c8355b68832c..a4d3c5ec614c 100644 --- a/packages/google-cloud-bigquery/tests/unit/job/test_query_job_retry.py +++ b/packages/google-cloud-bigquery/tests/unit/job/test_query_job_retry.py @@ -11,14 +11,13 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. +import concurrent.futures from unittest import mock -import concurrent.futures import freezegun -from google.api_core import exceptions import google.api_core.retry import pytest - +from google.api_core import exceptions from google.cloud.bigquery.client import _MIN_GET_QUERY_RESULTS_TIMEOUT from google.cloud.bigquery.job import QueryJob from google.cloud.bigquery.retry import DEFAULT_GET_JOB_TIMEOUT @@ -27,7 +26,6 @@ from ..helpers import make_connection from .helpers import _make_client - PROJECT = "test-project" JOB_ID = "test-job-id" QUERY = "select count(*) from persons" @@ -105,7 +103,7 @@ def test_result_w_custom_retry(global_time_lock): initial=0.001, maximum=0.001, multiplier=1.0, - deadline=0.1, + deadline=1.0, predicate=custom_predicate, )