Added test for Postgres arrays with pg

Author: ankane

pg_test.go

@@ -19,6 +19,7 @@ type PgItem struct {
 	HalfEmbedding   pgvector.HalfVector   `pg:"type:halfvec(3)"`
 	BinaryEmbedding string                `pg:"type:bit(3)"`
 	SparseEmbedding pgvector.SparseVector `pg:"type:sparsevec(3)"`
+	Embeddings      []pgvector.Vector     `pg:"type:vector(3)[]"`
 }
 
 func CreatePgItems(db *pg.DB) {
@@ -28,18 +29,21 @@ func CreatePgItems(db *pg.DB) {
 			HalfEmbedding:   pgvector.NewHalfVector([]float32{1, 1, 1}),
 			BinaryEmbedding: "000",
 			SparseEmbedding: pgvector.NewSparseVector([]float32{1, 1, 1}),
+			Embeddings:      []pgvector.Vector{pgvector.NewVector([]float32{1, 1, 1})},
 		},
 		PgItem{
 			Embedding:       pgvector.NewVector([]float32{2, 2, 2}),
 			HalfEmbedding:   pgvector.NewHalfVector([]float32{2, 2, 2}),
 			BinaryEmbedding: "101",
 			SparseEmbedding: pgvector.NewSparseVector([]float32{2, 2, 2}),
+			Embeddings:      []pgvector.Vector{pgvector.NewVector([]float32{2, 2, 2})},
 		},
 		PgItem{
 			Embedding:       pgvector.NewVector([]float32{1, 1, 2}),
 			HalfEmbedding:   pgvector.NewHalfVector([]float32{1, 1, 2}),
 			BinaryEmbedding: "111",
 			SparseEmbedding: pgvector.NewSparseVector([]float32{1, 1, 2}),
+			Embeddings:      []pgvector.Vector{pgvector.NewVector([]float32{1, 1, 2})},
 		},
 	}
 
@@ -93,6 +97,9 @@ func TestPg(t *testing.T) {
 	if !reflect.DeepEqual(items[1].SparseEmbedding.Slice(), []float32{1, 1, 2}) {
 		t.Error()
 	}
+	if !reflect.DeepEqual(items[1].Embeddings, []pgvector.Vector{pgvector.NewVector([]float32{1, 1, 2})}) {
+		t.Error()
+	}
 
 	var distances []float64
 	err = db.Model(&items).ColumnExpr("embedding <-> ?", pgvector.NewVector([]float32{1, 1, 1})).Order("id").Select(&distances)