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) }