[vision] fix: mark a test as flaky

fixes GoogleCloudPlatform#3702
tmatsuo · May 7, 2020 · bbd4253 · bbd4253
1 parent 04bbc2a
commit bbd4253
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/vision/cloud-client/web/requirements-test.txt b/vision/cloud-client/web/requirements-test.txt
@@ -1 +1,2 @@
+flaky==3.6.1
 pytest==5.3.2
diff --git a/vision/cloud-client/web/web_detect_test.py b/vision/cloud-client/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))