update: add tests for decoding base64 encoded emeddings

Signed-off-by: Milos Gajdos <[email protected]>

embeddings_test.go

@@ -1,6 +1,9 @@
 package embeddings
 
-import "testing"
+import (
+	"reflect"
+	"testing"
+)
 
 func TestToFloat32(t *testing.T) {
 	e := Embedding{
@@ -20,3 +23,47 @@ func TestToFloat32(t *testing.T) {
 		}
 	}
 }
+
+func TestBase64Decode(t *testing.T) {
+	t.Parallel()
+	testCases := []struct {
+		name    string
+		given   string
+		exp     []float64
+		wantErr bool
+	}{
+		{
+			name:    "valid",
+			given:   "H4XrUbgeCUAAAAAAAABFwESLbOf7QI9A",
+			exp:     []float64{3.14, -42.0, 1000.123},
+			wantErr: false,
+		},
+		{
+			name:    "invalid",
+			given:   "garbage",
+			wantErr: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		tc := tc
+		t.Run(tc.name, func(t *testing.T) {
+			t.Parallel()
+			embBase64 := Base64(tc.given)
+			got, err := embBase64.Decode()
+			if err != nil {
+				if !tc.wantErr {
+					t.Fatalf("unexpected error: %v", err)
+				}
+				return
+			}
+			// no error, but we expect one
+			if tc.wantErr {
+				t.Fatal("expected error")
+			}
+			if !reflect.DeepEqual(got, tc.exp) {
+				t.Fatalf("expected: %v, got: %v", tc.exp, got)
+			}
+		})
+	}
+}