24eff21aa4
Default to photo name when search term is too short or on the stop list. Search full text index otherwise, which now include names of people (requires reindexing).
33 lines
779 B
Go
33 lines
779 B
Go
package entity
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestFaces_Embeddings(t *testing.T) {
|
|
m := FaceFixtures.Get("joe-biden")
|
|
m1 := FaceFixtures.Get("jane-doe")
|
|
r := Faces{m, m1}.Embeddings()
|
|
len1 := len(m.Embedding())
|
|
len2 := len(m1.Embedding())
|
|
assert.Equal(t, len1+len2, len(r[0])+len(r[1]))
|
|
}
|
|
|
|
func TestFaces_IDs(t *testing.T) {
|
|
m := FaceFixtures.Get("joe-biden")
|
|
m1 := FaceFixtures.Get("jane-doe")
|
|
r := Faces{m, m1}.IDs()
|
|
assert.Equal(t, []string{"VF7ANLDET2BKZNT4VQWJMMC6HBEFDOG6", "VF7ANLDET2BKZNT4VQWJMMC6HBEFDOG7"}, r)
|
|
}
|
|
|
|
func TestDeleteOrphanFaces(t *testing.T) {
|
|
t.Run("Success", func(t *testing.T) {
|
|
if count, err := DeleteOrphanFaces(); err != nil {
|
|
t.Fatal(err)
|
|
} else {
|
|
t.Logf("deleted %d faces", count)
|
|
}
|
|
})
|
|
}
|