From 6d319da2c9f91bfc5f6573763a681a77fd522c23 Mon Sep 17 00:00:00 2001
From: Rui Wang Napieralski <wru@amazon.com>
Date: Tue, 16 Feb 2021 09:54:17 -0800
Subject: [PATCH] change: mark top 10 slow tests

---
 tests/integ/test_auto_ml.py               | 1 +
 tests/integ/test_clarify_model_monitor.py | 2 ++
 tests/integ/test_inference_pipeline.py    | 1 +
 tests/integ/test_model_quality_monitor.py | 1 +
 tests/integ/test_multidatamodel.py        | 2 ++
 tests/integ/test_mxnet.py                 | 2 ++
 tests/integ/test_tuner.py                 | 1 +
 7 files changed, 10 insertions(+)

diff --git a/tests/integ/test_auto_ml.py b/tests/integ/test_auto_ml.py
index 1409709aa7..6644dca12f 100644
--- a/tests/integ/test_auto_ml.py
+++ b/tests/integ/test_auto_ml.py
@@ -47,6 +47,7 @@
 }
 
 
+@pytest.mark.slow_test
 @pytest.mark.skipif(
     tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS,
     reason="AutoML is not supported in the region yet.",
diff --git a/tests/integ/test_clarify_model_monitor.py b/tests/integ/test_clarify_model_monitor.py
index 005cd50cce..6b2e69f653 100644
--- a/tests/integ/test_clarify_model_monitor.py
+++ b/tests/integ/test_clarify_model_monitor.py
@@ -285,6 +285,7 @@ def test_bias_monitor(sagemaker_session, scheduled_bias_monitor, endpoint_name,
     monitor.delete_monitoring_schedule()
 
 
+@pytest.mark.slow_test
 @pytest.mark.skipif(
     tests.integ.test_region() in tests.integ.NO_MODEL_MONITORING_REGIONS,
     reason="ModelMonitoring is not yet supported in this region.",
@@ -393,6 +394,7 @@ def test_explainability_monitor(sagemaker_session, scheduled_explainability_moni
     monitor.delete_monitoring_schedule()
 
 
+@pytest.mark.slow_test
 @pytest.mark.skipif(
     tests.integ.test_region() in tests.integ.NO_MODEL_MONITORING_REGIONS,
     reason="ModelMonitoring is not yet supported in this region.",
diff --git a/tests/integ/test_inference_pipeline.py b/tests/integ/test_inference_pipeline.py
index 016b8d313f..d086103195 100644
--- a/tests/integ/test_inference_pipeline.py
+++ b/tests/integ/test_inference_pipeline.py
@@ -149,6 +149,7 @@ def test_inference_pipeline_model_deploy(sagemaker_session, cpu_instance_type):
         assert "Could not find model" in str(exception.value)
 
 
+@pytest.mark.slow_test
 def test_inference_pipeline_model_deploy_and_update_endpoint(
     sagemaker_session, cpu_instance_type, alternative_cpu_instance_type
 ):
diff --git a/tests/integ/test_model_quality_monitor.py b/tests/integ/test_model_quality_monitor.py
index 9d18e3e070..fe41a40ab6 100644
--- a/tests/integ/test_model_quality_monitor.py
+++ b/tests/integ/test_model_quality_monitor.py
@@ -254,6 +254,7 @@ def test_run_model_quality_monitor(
     monitor.delete_monitoring_schedule()
 
 
+@pytest.mark.slow_test
 @pytest.mark.skipif(
     tests.integ.test_region() in tests.integ.NO_MODEL_MONITORING_REGIONS,
     reason="ModelMonitoring is not yet supported in this region.",
diff --git a/tests/integ/test_multidatamodel.py b/tests/integ/test_multidatamodel.py
index a853831a6b..b8b7dec42f 100644
--- a/tests/integ/test_multidatamodel.py
+++ b/tests/integ/test_multidatamodel.py
@@ -262,6 +262,7 @@ def test_multi_data_model_deploy_pretrained_models_local_mode(container_image, s
         assert "Could not find endpoint" in str(exception.value)
 
 
+@pytest.mark.slow_test
 @pytest.mark.skipif(
     tests.integ.test_region() != "us-east-2",
     reason="Pulling the base image is currently limited to us-east-2.",
@@ -481,6 +482,7 @@ def __rcf_training_job(
         return rcf_model
 
 
+@pytest.mark.slow_test
 @pytest.mark.skipif(
     tests.integ.test_region() != "us-east-2",
     reason="Pulling the base image is currently limited to us-east-2.",
diff --git a/tests/integ/test_mxnet.py b/tests/integ/test_mxnet.py
index 36e4121b72..e164596f89 100644
--- a/tests/integ/test_mxnet.py
+++ b/tests/integ/test_mxnet.py
@@ -83,6 +83,7 @@ def test_attach_deploy(mxnet_training_job, sagemaker_session, cpu_instance_type)
         assert result is not None
 
 
+@pytest.mark.slow_test
 def test_deploy_estimator_with_different_instance_types(
     mxnet_training_job,
     sagemaker_session,
@@ -258,6 +259,7 @@ def test_deploy_model_with_tags_and_kms(
         assert endpoint_config["KmsKeyId"] == kms_key_arn
 
 
+@pytest.mark.slow_test
 def test_deploy_model_and_update_endpoint(
     mxnet_training_job,
     sagemaker_session,
diff --git a/tests/integ/test_tuner.py b/tests/integ/test_tuner.py
index cc6a893567..b3287dfd3d 100644
--- a/tests/integ/test_tuner.py
+++ b/tests/integ/test_tuner.py
@@ -523,6 +523,7 @@ def test_stop_tuning_job(sagemaker_session, cpu_instance_type):
     assert desc["HyperParameterTuningJobStatus"] == "Stopping"
 
 
+@pytest.mark.slow_test
 @pytest.mark.canary_quick
 def test_tuning_mxnet(
     sagemaker_session,