From c2ac5fcdda2ebd8d7062d6044fcdb7745973cacd Mon Sep 17 00:00:00 2001 From: Takashi Matsuo Date: Mon, 6 Apr 2020 21:13:32 -0700 Subject: [PATCH] testing: mark some tests as flaky fixes #3138 --- datalabeling/manage_dataset_test.py | 4 ++++ datalabeling/requirements-test.txt | 1 + 2 files changed, 5 insertions(+) diff --git a/datalabeling/manage_dataset_test.py b/datalabeling/manage_dataset_test.py index ac7cd83fae58..02f3c9345776 100644 --- a/datalabeling/manage_dataset_test.py +++ b/datalabeling/manage_dataset_test.py @@ -33,6 +33,7 @@ def dataset(): manage_dataset.delete_dataset(dataset.name) +@pytest.mark.flaky(max_runs=3) def test_create_dataset(capsys): response = manage_dataset.create_dataset(PROJECT_ID) out, _ = capsys.readouterr() @@ -42,18 +43,21 @@ def test_create_dataset(capsys): manage_dataset.delete_dataset(response.name) +@pytest.mark.flaky(max_runs=3) def test_list_dataset(capsys, dataset): manage_dataset.list_datasets(PROJECT_ID) out, _ = capsys.readouterr() assert dataset.name in out +@pytest.mark.flaky(max_runs=3) def test_get_dataset(capsys, dataset): manage_dataset.get_dataset(dataset.name) out, _ = capsys.readouterr() assert "The dataset resource name:" in out +@pytest.mark.flaky(max_runs=3) def test_delete_dataset(capsys): # Creates a dataset. response = manage_dataset.create_dataset(PROJECT_ID) diff --git a/datalabeling/requirements-test.txt b/datalabeling/requirements-test.txt index 781d4326c947..1b569cb4f2c7 100644 --- a/datalabeling/requirements-test.txt +++ b/datalabeling/requirements-test.txt @@ -1 +1,2 @@ pytest==5.3.2 +flaky==3.6.1