diff --git a/packages/bigframes/tests/system/large/streaming/test_bigtable.py b/packages/bigframes/tests/system/large/streaming/test_bigtable.py index f10c534404e4..b76e5f7ae9d2 100644 --- a/packages/bigframes/tests/system/large/streaming/test_bigtable.py +++ b/packages/bigframes/tests/system/large/streaming/test_bigtable.py @@ -14,7 +14,7 @@ import time import uuid -from datetime import datetime, timedelta +from datetime import datetime, timedelta, timezone from typing import Generator import pytest @@ -68,7 +68,7 @@ def bigtable_table( bt_table.delete() -@pytest.mark.flaky(retries=3, delay=10) +@pytest.mark.flaky(retries=3, delay=30) def test_streaming_df_to_bigtable( session_load: bigframes.Session, bigtable_table: table.Table ): @@ -92,7 +92,7 @@ def test_streaming_df_to_bigtable( bigtable_options={}, job_id=None, job_id_prefix=job_id_prefix, - start_timestamp=datetime.now() - timedelta(days=1), + start_timestamp=datetime.now(timezone.utc) - timedelta(days=1), ) # wait 200 seconds in order to ensure the query doesn't stop diff --git a/packages/bigframes/tests/system/large/streaming/test_pubsub.py b/packages/bigframes/tests/system/large/streaming/test_pubsub.py index cdc27ae65cf8..d4e5a9fd2aa9 100644 --- a/packages/bigframes/tests/system/large/streaming/test_pubsub.py +++ b/packages/bigframes/tests/system/large/streaming/test_pubsub.py @@ -14,7 +14,7 @@ import uuid from concurrent import futures -from datetime import datetime, timedelta +from datetime import datetime, timedelta, timezone from typing import Generator import pytest @@ -59,7 +59,7 @@ def pubsub_topic_subscription_ids( subscriber.delete_subscription(subscription=subscription_name) -@pytest.mark.flaky(retries=3, delay=10) +@pytest.mark.flaky(retries=3, delay=30) def test_streaming_df_to_pubsub( session_load: bigframes.Session, pubsub_topic_subscription_ids: tuple[str, str] ): @@ -100,7 +100,7 @@ def callback(message): service_account_email="streaming-testing@bigframes-load-testing.iam.gserviceaccount.com", job_id=None, job_id_prefix=job_id_prefix, - start_timestamp=datetime.now() - timedelta(days=1), + start_timestamp=datetime.now(timezone.utc) - timedelta(days=1), ) try: # wait 200 seconds in order to ensure the query doesn't stop