From 77e8684252e958300d97cf852ce45deece04b331 Mon Sep 17 00:00:00 2001 From: Courtney Cochrane Date: Tue, 30 Jul 2019 11:21:32 -0400 Subject: [PATCH] Change pytest.mark.notebooks --- tests/integration/test_notebooks_question_answering.py | 6 ++++-- tests/integration/test_notebooks_sentence_similarity.py | 6 ++++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/integration/test_notebooks_question_answering.py b/tests/integration/test_notebooks_question_answering.py index 0538930..d333551 100644 --- a/tests/integration/test_notebooks_question_answering.py +++ b/tests/integration/test_notebooks_question_answering.py @@ -9,7 +9,8 @@ from tests.notebooks_common import OUTPUT_NOTEBOOK ABS_TOL = 0.2 -@pytest.mark.notebooks +@pytest.mark.integration +@pytest.mark.azureml def test_bidaf_deep_dive(notebooks, subscription_id, resource_group, @@ -32,7 +33,8 @@ def test_bidaf_deep_dive(notebooks, assert result == pytest.approx(0.5, abs=ABS_TOL) @pytest.mark.usefixtures("teardown_service") -@pytest.mark.notebooks +@pytest.mark.integration +@pytest.mark.azureml def test_bidaf_quickstart(notebooks, subscription_id, resource_group, diff --git a/tests/integration/test_notebooks_sentence_similarity.py b/tests/integration/test_notebooks_sentence_similarity.py index 4b0215a..c83ed9f 100644 --- a/tests/integration/test_notebooks_sentence_similarity.py +++ b/tests/integration/test_notebooks_sentence_similarity.py @@ -36,7 +36,8 @@ def baseline_results(): "Doc2vec Cosine with Stop Words": 0.4498543211602068, } -@pytest.mark.notebooks +@pytest.mark.integration +@pytest.mark.azureml def test_similarity_embeddings_baseline_runs(notebooks, baseline_results): notebook_path = notebooks["similarity_embeddings_baseline"] pm.execute_notebook(notebook_path, OUTPUT_NOTEBOOK) @@ -45,7 +46,8 @@ def test_similarity_embeddings_baseline_runs(notebooks, baseline_results): assert results[key] == pytest.approx(value, abs=ABS_TOL) @pytest.mark.usefixtures("teardown_service") -@pytest.mark.notebooks +@pytest.mark.integration +@pytest.mark.azureml def test_automl_local_runs(notebooks, subscription_id, resource_group,