From 36d9ec72594426923abe0de8982de918a834349c Mon Sep 17 00:00:00 2001 From: Jeffrey Rennie Date: Mon, 11 Jun 2018 16:36:10 -0700 Subject: [PATCH] Fix the vision geo test. Sometimes, Vision sees Zepra. Othertimes, it sees Electra Tower. --- vision/cloud-client/detect/detect_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vision/cloud-client/detect/detect_test.py b/vision/cloud-client/detect/detect_test.py index 0510d1006d5..71be19440fb 100644 --- a/vision/cloud-client/detect/detect_test.py +++ b/vision/cloud-client/detect/detect_test.py @@ -217,14 +217,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):