chore: fix flaky test by adding rerun (#77)

* chore: fix flaky test by adding rerun * fixed the lint
googleapis · Dec 2, 2020 · 23e8599 · 23e8599
1 parent 6679033
commit 23e8599
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/packages/google-cloud-vision/samples/snippets/detect/detect_test.py b/packages/google-cloud-vision/samples/snippets/detect/detect_test.py
@@ -17,6 +17,7 @@
 
 import backoff
 from google.cloud import storage
+import pytest
 
 import detect
 
@@ -242,6 +243,7 @@ def test_detect_crop_hints_uri(capsys):
     assert 'bounds: ' in out
 
 
+@pytest.mark.flaky(max_runs=3, min_passes=1)
 def test_async_detect_document(capsys):
     storage_client = storage.Client()
     bucket = storage_client.get_bucket(BUCKET)