From a074ec5c3d0f3afbd4c0fef46b097eadcc8c6c67 Mon Sep 17 00:00:00 2001 From: Yu-Han Liu Date: Wed, 18 Apr 2018 10:45:45 -0700 Subject: [PATCH 1/2] detect-pdf update --- vision/cloud-client/detect/detect_pdf_test.py | 6 ++++-- vision/cloud-client/detect/requirements.txt | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/vision/cloud-client/detect/detect_pdf_test.py b/vision/cloud-client/detect/detect_pdf_test.py index f0f0b5f7126..2cfbc0787cf 100644 --- a/vision/cloud-client/detect/detect_pdf_test.py +++ b/vision/cloud-client/detect/detect_pdf_test.py @@ -25,6 +25,10 @@ def test_async_detect_document(capsys): + storage_client = storage.Client() + bucket = storage_client.get_bucket(BUCKET) + assert len(list(bucket.list_blobs(prefix=OUTPUT_PREFIX))) == 0 + async_detect_document( gcs_source_uri=GCS_SOURCE_URI, gcs_destination_uri=GCS_DESTINATION_URI) @@ -32,7 +36,5 @@ def test_async_detect_document(capsys): assert 'Hodge conjecture' in out - storage_client = storage.Client() - bucket = storage_client.get_bucket(BUCKET) for blob in bucket.list_blobs(prefix=OUTPUT_PREFIX): blob.delete() diff --git a/vision/cloud-client/detect/requirements.txt b/vision/cloud-client/detect/requirements.txt index ca9d0e400b8..a6c769cef90 100644 --- a/vision/cloud-client/detect/requirements.txt +++ b/vision/cloud-client/detect/requirements.txt @@ -1,2 +1,2 @@ -google-cloud-vision==0.30.1 +google-cloud-vision==0.31.0 google-cloud-storage==1.6.0 From b50698511879e93df3bafc997f659eb576ad67db Mon Sep 17 00:00:00 2001 From: Yu-Han Liu Date: Wed, 18 Apr 2018 10:50:37 -0700 Subject: [PATCH 2/2] update test --- vision/cloud-client/detect/detect_pdf_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vision/cloud-client/detect/detect_pdf_test.py b/vision/cloud-client/detect/detect_pdf_test.py index 2cfbc0787cf..98b06aaab74 100644 --- a/vision/cloud-client/detect/detect_pdf_test.py +++ b/vision/cloud-client/detect/detect_pdf_test.py @@ -35,6 +35,7 @@ def test_async_detect_document(capsys): out, _ = capsys.readouterr() assert 'Hodge conjecture' in out + assert len(list(bucket.list_blobs(prefix=OUTPUT_PREFIX))) == 3 for blob in bucket.list_blobs(prefix=OUTPUT_PREFIX): blob.delete()