[vision] testing: re-enable test_async_detect_document

fixes GoogleCloudPlatform#3753 also made the data PDF to be smaller.
tmatsuo · May 14, 2020 · 75a4f00 · 75a4f00
1 parent ec30b6a
commit 75a4f00
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/vision/cloud-client/detect/detect_test.py b/vision/cloud-client/detect/detect_test.py
@@ -16,7 +16,6 @@
 import uuid
 
 from google.cloud import storage
-import pytest
 
 import detect
 
@@ -209,7 +208,6 @@ def test_detect_crop_hints_uri(capsys):
     assert 'bounds: ' in out
 
 
-@pytest.mark.skip("It's constantly failing.")
 def test_async_detect_document(capsys):
     storage_client = storage.Client()
     bucket = storage_client.get_bucket(BUCKET)
@@ -219,7 +217,8 @@ def test_async_detect_document(capsys):
 
     assert len(list(bucket.list_blobs(prefix=OUTPUT_PREFIX))) == 0
 
-    uri = 'gs://{}/vision/document/custom_0773375000.pdf'.format(ASSET_BUCKET)
+    uri = 'gs://{}/vision/document/custom_0773375000_title_only.pdf'.format(
+        ASSET_BUCKET)
     detect.async_detect_document(
         gcs_source_uri=uri,
         gcs_destination_uri=GCS_DESTINATION_URI)

diff --git a/...ient/detect/resources/cloud-samples-data/vision/document/custom_0773375000_title_only.pdf b/...ient/detect/resources/cloud-samples-data/vision/document/custom_0773375000_title_only.pdf