Fix the vision geo test. (#1518)

Sometimes, Vision sees Zepra. Othertimes, it sees Electra Tower.
GoogleCloudPlatform · Jun 22, 2018 · e0b0a00 · e0b0a00
1 parent 88fef98
commit e0b0a00
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
@@ -222,14 +222,14 @@ def test_detect_web_with_geo(capsys):
         'resources/city.jpg')
     detect.web_entities_include_geo_results(file_name)
     out, _ = capsys.readouterr()
-    assert 'Zepra' in out
+    assert 'Zepra' in out or 'Electra Tower' in out
 
 
 def test_detect_web_with_geo_uri(capsys):
     file_name = 'gs://{}/vision/city.jpg'.format(BUCKET)
     detect.web_entities_include_geo_results_uri(file_name)
     out, _ = capsys.readouterr()
-    assert 'Zepra' in out
+    assert 'Zepra' in out or 'Electra Tower' in out
 
 
 def test_detect_document(capsys):