Mark tests for continuous testing

Author: laurenyu

tests/integ/test_byo_estimator.py

@@ -42,6 +42,7 @@ def fm_serializer(data):
     return json.dumps(js)
 
 
+@pytest.mark.continuous_testing
 def test_byo_estimator(sagemaker_session, region):
     """Use Factorization Machines algorithm as an example here.
 

tests/integ/test_factorization_machines.py

@@ -16,12 +16,15 @@
 import sys
 import time
 
+import pytest
+
 from sagemaker import FactorizationMachines, FactorizationMachinesModel
 from sagemaker.utils import name_from_base
 from tests.integ import DATA_DIR
 from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
 
 
+@pytest.mark.continuous_testing
 def test_factorization_machines(sagemaker_session):
     with timeout(minutes=15):
         data_path = os.path.join(DATA_DIR, 'one_p_mnist', 'mnist.pkl.gz')

tests/integ/test_kmeans.py

@@ -16,12 +16,15 @@
 import sys
 import time
 
+import pytest
+
 from sagemaker import KMeans, KMeansModel
 from sagemaker.utils import name_from_base
 from tests.integ import DATA_DIR
 from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
 
 
+@pytest.mark.continuous_testing
 def test_kmeans(sagemaker_session):
     with timeout(minutes=15):
         data_path = os.path.join(DATA_DIR, 'one_p_mnist', 'mnist.pkl.gz')

tests/integ/test_lda.py

@@ -13,6 +13,7 @@
 import os
 
 import numpy as np
+import pytest
 
 from sagemaker import LDA, LDAModel
 from sagemaker.amazon.common import read_records
@@ -22,6 +23,7 @@
 from tests.integ.record_set import prepare_record_set_from_local_files
 
 
+@pytest.mark.continuous_testing
 def test_lda(sagemaker_session):
     with timeout(minutes=15):
         data_path = os.path.join(DATA_DIR, 'lda')

tests/integ/test_linear_learner.py

@@ -17,13 +17,15 @@
 import time
 
 import numpy as np
+import pytest
 
 from sagemaker.amazon.linear_learner import LinearLearner, LinearLearnerModel
 from sagemaker.utils import name_from_base, sagemaker_timestamp
 from tests.integ import DATA_DIR
 from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
 
 
+@pytest.mark.continuous_testing
 def test_linear_learner(sagemaker_session):
     with timeout(minutes=15):
         data_path = os.path.join(DATA_DIR, 'one_p_mnist', 'mnist.pkl.gz')

tests/integ/test_mxnet_train.py

@@ -52,6 +52,7 @@ def test_attach_deploy(mxnet_training_job, sagemaker_session):
         predictor.predict(data)
 
 
+@pytest.mark.continuous_testing
 def test_deploy_model(mxnet_training_job, sagemaker_session):
     endpoint_name = 'test-mxnet-deploy-model-{}'.format(sagemaker_timestamp())
 

tests/integ/test_ntm.py

@@ -13,6 +13,7 @@
 import os
 
 import numpy as np
+import pytest
 
 from sagemaker import NTM, NTMModel
 from sagemaker.amazon.common import read_records
@@ -22,6 +23,7 @@
 from tests.integ.record_set import prepare_record_set_from_local_files
 
 
+@pytest.mark.continuous_testing
 def test_ntm(sagemaker_session):
     with timeout(minutes=15):
         data_path = os.path.join(DATA_DIR, 'ntm')

tests/integ/test_pca.py

@@ -16,12 +16,15 @@
 import sys
 import time
 
+import pytest
+
 import sagemaker.amazon.pca
 from sagemaker.utils import name_from_base
 from tests.integ import DATA_DIR
 from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
 
 
+@pytest.mark.continuous_testing
 def test_pca(sagemaker_session):
     with timeout(minutes=15):
         data_path = os.path.join(DATA_DIR, 'one_p_mnist', 'mnist.pkl.gz')

tests/integ/test_randomcutforest.py

@@ -11,12 +11,14 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 import numpy as np
+import pytest
 
 from sagemaker import RandomCutForest, RandomCutForestModel
 from sagemaker.utils import name_from_base
 from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
 
 
+@pytest.mark.continuous_testing
 def test_randomcutforest(sagemaker_session):
     with timeout(minutes=15):
         # Generate a thousand 14-dimensional datapoints.

tests/integ/test_tf.py

@@ -22,6 +22,7 @@
 DATA_PATH = os.path.join(DATA_DIR, 'iris', 'data')
 
 
+@pytest.mark.continuous_testing
 def test_tf(sagemaker_session, tf_full_version):
     with timeout(minutes=15):
         script_path = os.path.join(DATA_DIR, 'iris', 'iris-dnn-classifier.py')