[vision] fix: mark a test as flaky [(#3709)](GoogleCloudPlatform/pyth…

…on-docs-samples#3709) fixes #3702
busunkim96 · May 8, 2020 · 893b31b · 893b31b
1 parent 974370a
commit 893b31b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/samples/snippets/web/requirements-test.txt b/samples/snippets/web/requirements-test.txt
@@ -1 +1,2 @@
+flaky==3.6.1
 pytest==5.3.2
diff --git a/samples/snippets/web/web_detect_test.py b/samples/snippets/web/web_detect_test.py
@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import pytest
+
 import web_detect
 
 ASSET_BUCKET = "cloud-samples-data"
@@ -24,6 +26,7 @@ def test_detect_file(capsys):
     assert 'description' in out.lower()
 
 
+@pytest.mark.flaky(max_runs=3, min_passes=1)
 def test_detect_web_gsuri(capsys):
     file_name = ('gs://{}/vision/landmark/pofa.jpg'.format(
                  ASSET_BUCKET))