chore: import upstream snapshot with attribution

internal/embedding/validate_test.go

@@ -0,0 +1,33 @@
+package embedding
+
+import "testing"
+
+func TestValidateBatch(t *testing.T) {
+	vec := func(n int) []float32 { return make([]float32, n) }
+
+	cases := []struct {
+		name    string
+		texts   []string
+		vecs    [][]float32
+		dims    int
+		wantErr bool
+	}{
+		{"ok strict width", []string{"a", "b"}, [][]float32{vec(384), vec(384)}, 384, false},
+		{"ok empty batch", []string{}, [][]float32{}, 384, false},
+		{"ok lazy width consistent", []string{"a", "b"}, [][]float32{vec(768), vec(768)}, 0, false},
+		{"count mismatch short", []string{"a", "b"}, [][]float32{vec(384)}, 384, true},
+		{"count mismatch extra", []string{"a"}, [][]float32{vec(384), vec(384)}, 384, true},
+		{"nil vector", []string{"a", "b"}, [][]float32{vec(384), nil}, 384, true},
+		{"empty vector", []string{"a"}, [][]float32{{}}, 384, true},
+		{"wrong strict width", []string{"a"}, [][]float32{vec(100)}, 384, true},
+		{"ragged lazy width", []string{"a", "b"}, [][]float32{vec(768), vec(384)}, 0, true},
+	}
+	for _, tc := range cases {
+		t.Run(tc.name, func(t *testing.T) {
+			err := validateBatch("test", tc.texts, tc.vecs, tc.dims)
+			if (err != nil) != tc.wantErr {
+				t.Fatalf("validateBatch err = %v, wantErr %v", err, tc.wantErr)
+			}
+		})
+	}
+}