vision geo test fix (#2353)

Gus already LGTM
GoogleCloudPlatform · Aug 26, 2019 · b88baf9 · b88baf9
1 parent 6996c8e
commit b88baf9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vision/cloud-client/detect/detect_test.py b/vision/cloud-client/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):