Skip to content

Commit dc67b74

Browse files
shollymanshollyman
authored
vision: update samples to address changes in model annotations. (GoogleCloudPlatform#1991)
changes to the vision model evaluation changed annotations for some of the sample data used in these tests. This corrects those expectations to reflect current evaluation. Background: internal issue 123358697
1 parent 2bc5cef commit dc67b74

2 files changed

Lines changed: 3 additions & 6 deletions

File tree

‎vision/cloud-client/detect/detect_test.py‎

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -100,21 +100,21 @@ def test_logos(capsys):
100100
'resources/logos.png')
101101
detect.detect_logos(file_name)
102102
out, _ = capsys.readouterr()
103-
assert 'Google' in out
103+
assert 'google' in out
104104

105105

106106
def test_logos_uri(capsys):
107107
file_name = 'gs://{}/vision/logos.png'.format(BUCKET)
108108
detect.detect_logos_uri(file_name)
109109
out, _ = capsys.readouterr()
110-
assert 'Google' in out
110+
assert 'google' in out
111111

112112

113113
def test_logos_http(capsys):
114114
uri = 'https://storage-download.googleapis.com/{}/vision/logos.png'
115115
detect.detect_logos_uri(uri.format(BUCKET))
116116
out, _ = capsys.readouterr()
117-
assert 'Google' in out
117+
assert 'google' in out
118118

119119

120120
def test_safe_search(capsys):
@@ -196,23 +196,20 @@ def test_detect_web(capsys):
196196
'resources/landmark.jpg')
197197
detect.detect_web(file_name)
198198
out, _ = capsys.readouterr()
199-
assert 'Description: Palace of Fine Arts Theatre' in out
200199
assert 'Best guess label: palace of fine arts' in out
201200

202201

203202
def test_detect_web_uri(capsys):
204203
file_name = 'gs://{}/vision/landmark.jpg'.format(BUCKET)
205204
detect.detect_web_uri(file_name)
206205
out, _ = capsys.readouterr()
207-
assert 'Description: Palace of Fine Arts Theatre' in out
208206
assert 'Best guess label: palace of fine arts' in out
209207

210208

211209
def test_detect_web_http(capsys):
212210
uri = 'https://storage-download.googleapis.com/{}/vision/landmark.jpg'
213211
detect.detect_web_uri(uri.format(BUCKET))
214212
out, _ = capsys.readouterr()
215-
assert 'Description: Palace of Fine Arts Theatre' in out
216213
assert 'Best guess label: palace of fine arts' in out
217214

218215

‎vision/cloud-client/detect/resources/logos.png‎

13 KB
Loading

0 commit comments

Comments
 (0)