vision geo test fix [(#2353)](GoogleCloudPlatform/python-docs-samples…

…#2353) Gus already LGTM
busunkim96 · Aug 26, 2019 · bb57d0c · bb57d0c
1 parent 56ea938
commit bb57d0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/samples/snippets/detect/detect_test.py b/samples/snippets/detect/detect_test.py
@@ -225,15 +225,15 @@ def test_detect_web_with_geo(capsys):
     detect.web_entities_include_geo_results(file_name)
     out, _ = capsys.readouterr()
     out = out.lower()
-    assert 'zepra' in out or 'electra tower' in out
+    assert 'zepra' in out or 'electra tower' in out or 'tel aviv' in out
 
 
 def test_detect_web_with_geo_uri(capsys):
     file_name = 'gs://{}/vision/web/city.jpg'.format(ASSET_BUCKET)
     detect.web_entities_include_geo_results_uri(file_name)
     out, _ = capsys.readouterr()
     out = out.lower()
-    assert 'zepra' in out or 'electra tower' in out
+    assert 'zepra' in out or 'electra tower' in out or 'tel aviv' in out
 
 
 def test_detect_document(capsys):