8000 Mark tests as slow · mmmarklu/python-docs-samples@1752d90 · GitHub
[go: up one dir, main page]
Skip to content

Commit 1752d90

Browse files
nnegreynnegrey
committed
Mark tests as slow
1 parent 93b8d4f commit 1752d90

8 files changed

+14
-0
lines changed

‎automl/cloud-client/language_entity_extraction_dataset_management_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ def test_get_dataset(capsys):
6767
assert 'Dataset name: ' in out
6868

6969

70+
@pytest.mark.slow
7071
def test_export_dataset(capsys):
7172
export_dataset.export_dataset(
7273
PROJECT_ID,

‎automl/cloud-client/language_entity_extraction_predict_test.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616

1717
import os
1818

19+
import pytest
20+
1921
import language_batch_predict
2022
import language_entity_extraction_predict
2123

@@ -32,6 +34,7 @@ def test_predict(capsys):
3234
assert 'Text Extract Entity Types: ' in out
3335

3436

37+
@pytest.mark.slow
3538
def test_batch_predict(capsys):
3639
model_id = 'TEN5112482778553778176'
3740
input_uri = 'gs://{}/entity_extraction/input.jsonl'.format(BUCKET_ID)

‎automl/cloud-client/language_sentiment_analysis_dataset_management_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ def test_get_dataset(capsys):
6767
assert 'Dataset name: ' in out
6868

6969

70+
@pytest.mark.slow
7071
def test_export_dataset(capsys):
7172
export_dataset.export_dataset(
7273
PROJECT_ID,

‎automl/cloud-client/language_text_classification_dataset_management_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ def test_get_dataset(capsys):
6666
assert 'Dataset name: ' in out
6767

6868

69+
@pytest.mark.slow
6970
def test_export_dataset(capsys):
7071
export_dataset.export_dataset(
7172
PROJECT_ID,

‎automl/cloud-client/vision_classification_dataset_management_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ def test_get_dataset(capsys):
6666
assert 'Dataset name: ' in out
6767

6868

69+
@pytest.mark.slow
6970
def test_export_dataset(capsys):
7071
export_dataset.export_dataset(
7172
PROJECT_ID,

‎automl/cloud-client/vision_classification_predict_test.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616

1717
import os
1818

19+
import pytest
20+
1921
import vision_batch_predict
2022
import vision_classification_predict
2123

@@ -31,6 +33,7 @@ def test_predict(capsys):
3133
assert 'Predicted class name:' in out
3234

3335

36+
@pytest.mark.slow
3437
def test_batch_predict(capsys):
3538
model_id = 'ICN7383667271543079510'
3639
input_uri = 'gs://{}/batch_predict_test.csv'.format(BUCKET_ID)

‎automl/cloud-client/vision_object_detection_dataset_management_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ def test_get_dataset(capsys):
6666
assert 'Dataset name: ' in out
6767

6868

69+
@pytest.mark.slow
6970
def test_export_dataset(capsys):
7071
export_dataset.export_dataset(
7172
PROJECT_ID,

‎automl/cloud-client/vision_object_detection_predict_test.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616

1717
import os
1818

19+
import pytest
20+
1921
import vision_batch_predict
2022
import vision_object_detection_predict
2123

@@ -32,6 +34,7 @@ def test_predict(capsys):
3234
assert 'Y:' in out
3335

3436

37+
@pytest.mark.slow
3538
def test_batch_predict(capsys):
3639
model_id = 'IOD6143103405779845120'
3740
input_uri = 'gs://{}/vision_object_detection_batch_predict_test.csv'\

0 commit comments

Comments
 (0)
0