photoprism/internal/entity/embedding_test.go

81 lines
1.9 KiB
Go
Raw Normal View History

2021-08-25 17:54:06 +02:00
package entity
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestEmbeddingsMidpoint(t *testing.T) {
t.Run("2 embeddings, 1 dimension", func(t *testing.T) {
e := Embeddings{Embedding{1}, Embedding{3}}
_, r, c := EmbeddingsMidpoint(e)
assert.Equal(t, 1.01, r)
assert.Equal(t, 2, c)
})
t.Run("3 embeddings, 1 dimension", func(t *testing.T) {
e := Embeddings{Embedding{1}, Embedding{3}, Embedding{4}}
_, r, c := EmbeddingsMidpoint(e)
assert.Equal(t, 1.01, r)
assert.Equal(t, 3, c)
})
t.Run("4 embeddings, 1 dimension", func(t *testing.T) {
e := Embeddings{Embedding{1}, Embedding{3}, Embedding{4}, Embedding{8}}
_, r, c := EmbeddingsMidpoint(e)
assert.Equal(t, 2.51, r)
assert.Equal(t, 4, c)
})
t.Run("empty embedding", func(t *testing.T) {
e := Embeddings{}
_, r, c := EmbeddingsMidpoint(e)
assert.Equal(t, float64(0), r)
assert.Equal(t, 0, c)
})
t.Run("embedding with different length", func(t *testing.T) {
e := Embeddings{Embedding{1}, Embedding{3, 5}}
_, r, c := EmbeddingsMidpoint(e)
assert.Equal(t, float64(0), r)
assert.Equal(t, 2, c)
})
}
func TestUnmarshalEmbedding(t *testing.T) {
t.Run("success", func(t *testing.T) {
r := UnmarshalEmbedding("[-0.013,-0.031]")
assert.Equal(t, []float64{-0.013, -0.031}, r)
})
t.Run("no prefix", func(t *testing.T) {
r := UnmarshalEmbedding("-0.013,-0.031]")
assert.Nil(t, r)
})
2021-08-26 11:58:52 +02:00
t.Run("invalid json", func(t *testing.T) {
r := UnmarshalEmbedding("[true, false]")
assert.Equal(t, []float64{0, 0}, r)
})
2021-08-25 17:54:06 +02:00
}
func TestUnmarshalEmbeddings(t *testing.T) {
t.Run("success", func(t *testing.T) {
r := UnmarshalEmbeddings("[[-0.013,-0.031]]")
assert.Equal(t, [][]float64{{-0.013, -0.031}}, r)
})
t.Run("no prefix", func(t *testing.T) {
r := UnmarshalEmbeddings("-0.013,-0.031]")
assert.Nil(t, r)
})
2021-08-26 11:58:52 +02:00
t.Run("invalid json", func(t *testing.T) {
r := UnmarshalEmbeddings("[[true, false]]")
assert.Equal(t, [][]float64{{0, 0}}, r)
})
2021-08-25 17:54:06 +02:00
}