This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new f67845acc2 make tests run faster (#27492)
f67845acc2 is described below
commit f67845acc2f0f317d52b9ee017e14e6dbaeb8317
Author: Raphaël Vandon <[email protected]>
AuthorDate: Fri Nov 4 04:48:33 2022 -0700
make tests run faster (#27492)
from 22s to 4s on my machine for all sensor tests
---
tests/providers/amazon/aws/sensors/test_sagemaker_endpoint.py | 2 +-
tests/providers/amazon/aws/sensors/test_sagemaker_training.py | 4 ++--
tests/providers/amazon/aws/sensors/test_sagemaker_transform.py | 2 +-
tests/providers/amazon/aws/sensors/test_sagemaker_tuning.py | 2 +-
4 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/tests/providers/amazon/aws/sensors/test_sagemaker_endpoint.py
b/tests/providers/amazon/aws/sensors/test_sagemaker_endpoint.py
index c46232110a..f71183be3a 100644
--- a/tests/providers/amazon/aws/sensors/test_sagemaker_endpoint.py
+++ b/tests/providers/amazon/aws/sensors/test_sagemaker_endpoint.py
@@ -79,7 +79,7 @@ class TestSageMakerEndpointSensor(unittest.TestCase):
DESCRIBE_ENDPOINT_INSERVICE_RESPONSE,
]
sensor = SageMakerEndpointSensor(
- task_id="test_task", poke_interval=1, aws_conn_id="aws_test",
endpoint_name="test_job_name"
+ task_id="test_task", poke_interval=0, aws_conn_id="aws_test",
endpoint_name="test_job_name"
)
sensor.execute(None)
diff --git a/tests/providers/amazon/aws/sensors/test_sagemaker_training.py
b/tests/providers/amazon/aws/sensors/test_sagemaker_training.py
index ba6b5e9c34..3a13384f25 100644
--- a/tests/providers/amazon/aws/sensors/test_sagemaker_training.py
+++ b/tests/providers/amazon/aws/sensors/test_sagemaker_training.py
@@ -80,7 +80,7 @@ class TestSageMakerTrainingSensor(unittest.TestCase):
]
sensor = SageMakerTrainingSensor(
task_id="test_task",
- poke_interval=2,
+ poke_interval=0,
aws_conn_id="aws_test",
job_name="test_job_name",
print_log=False,
@@ -113,7 +113,7 @@ class TestSageMakerTrainingSensor(unittest.TestCase):
]
sensor = SageMakerTrainingSensor(
task_id="test_task",
- poke_interval=2,
+ poke_interval=0,
aws_conn_id="aws_test",
job_name="test_job_name",
print_log=True,
diff --git a/tests/providers/amazon/aws/sensors/test_sagemaker_transform.py
b/tests/providers/amazon/aws/sensors/test_sagemaker_transform.py
index 8c962d5158..c6777165b2 100644
--- a/tests/providers/amazon/aws/sensors/test_sagemaker_transform.py
+++ b/tests/providers/amazon/aws/sensors/test_sagemaker_transform.py
@@ -77,7 +77,7 @@ class TestSageMakerTransformSensor(unittest.TestCase):
DESCRIBE_TRANSFORM_COMPLETED_RESPONSE,
]
sensor = SageMakerTransformSensor(
- task_id="test_task", poke_interval=2, aws_conn_id="aws_test",
job_name="test_job_name"
+ task_id="test_task", poke_interval=0, aws_conn_id="aws_test",
job_name="test_job_name"
)
sensor.execute(None)
diff --git a/tests/providers/amazon/aws/sensors/test_sagemaker_tuning.py
b/tests/providers/amazon/aws/sensors/test_sagemaker_tuning.py
index 735d8e1291..d7ff9153e4 100644
--- a/tests/providers/amazon/aws/sensors/test_sagemaker_tuning.py
+++ b/tests/providers/amazon/aws/sensors/test_sagemaker_tuning.py
@@ -80,7 +80,7 @@ class TestSageMakerTuningSensor(unittest.TestCase):
DESCRIBE_TUNING_COMPLETED_RESPONSE,
]
sensor = SageMakerTuningSensor(
- task_id="test_task", poke_interval=2, aws_conn_id="aws_test",
job_name="test_job_name"
+ task_id="test_task", poke_interval=0, aws_conn_id="aws_test",
job_name="test_job_name"
)
sensor.execute(None)
