photoprism/pkg/vector/vector_test.go
Michael Mayer 5738d838e5 Faces: Add dedicated vector algebra package #4691 #4669 #4328
Signed-off-by: Michael Mayer <michael@photoprism.app>
2025-02-03 16:48:46 +01:00

30 lines
754 B
Go

package vector
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestNewVector(t *testing.T) {
t.Run("Int", func(t *testing.T) {
v, err := NewVector([]int{1, 2, 3, 4, 6, 5})
assert.IsType(t, Vector{}, v)
assert.NoError(t, err)
})
t.Run("Float32", func(t *testing.T) {
v, err := NewVector([]float32{1.0, 2.1, 3.54, 4.9, 6.666666, 5.33333333})
assert.IsType(t, Vector{}, v)
assert.NoError(t, err)
})
t.Run("Float64", func(t *testing.T) {
v, err := NewVector([]float64{1.0, 2.1, 3.54, 4.9, 6.666666, 5.33333333})
assert.IsType(t, Vector{}, v)
assert.NoError(t, err)
})
t.Run("String", func(t *testing.T) {
v, err := NewVector([]string{"a", "b", "c"})
assert.IsType(t, Vector{}, v)
assert.Error(t, err)
})
}