samples: fixed flaky test (#57)

googleapis · Aug 31, 2020 · cd626e7 · cd626e7
1 parent 90879c5
commit cd626e7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/packages/google-cloud-translate/samples/snippets/hybrid_glossaries/hybrid_tutorial.py b/packages/google-cloud-translate/samples/snippets/hybrid_glossaries/hybrid_tutorial.py
@@ -56,6 +56,7 @@ def pic_to_text(infile):
     # For less dense text, use text_detection
     response = client.document_text_detection(image=image)
     text = response.full_text_annotation.text
+    print("Detected text: {}".format(text))
 
     return text
     # [END translate_hybrid_vision]

diff --git a/packages/google-cloud-translate/samples/snippets/hybrid_glossaries/hybrid_tutorial_test.py b/packages/google-cloud-translate/samples/snippets/hybrid_glossaries/hybrid_tutorial_test.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import os
-import re
 import sys
 import uuid
 
@@ -29,11 +28,13 @@
 # VISION TESTS
 
 
-def test_vision_standard_format():
+def test_vision_standard_format(capsys):
     # Generate text using Vision API
-    text = pic_to_text('resources/standard_format.jpeg')
+    pic_to_text('resources/standard_format.jpeg')
+    out, err = capsys.readouterr()
 
-    assert re.match(r"This\s?is\s?a\s?test!\s?", text)
+    assert 'Detected text:' in out
+    assert 'test!' in out
 
 
 def test_vision_non_standard_format():