Backend: Fix tensorflow_test.go

Signed-off-by: Michael Mayer <michael@liquidbytes.net>
This commit is contained in:
Michael Mayer 2019-12-15 18:36:27 +01:00
parent a4b475cbd6
commit 4b7857891a

View file

@ -233,7 +233,7 @@ func TestTensorFlow_BestLabels(t *testing.T) {
p[1] = 0.5
result := tensorFlow.bestLabels(p)
assert.Equal(t, "hen", result[0].Name)
assert.Equal(t, "chicken", result[0].Name)
assert.Equal(t, "bird", result[0].Categories[0])
assert.Equal(t, "image", result[0].Source)
assert.Equal(t, "goldfish", result[1].Name)