Added tests for String method

pgvector · Jun 4, 2024 · ef87dad · ef87dad
1 parent e41d95f
commit ef87dad
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/halfvec_test.go b/halfvec_test.go
@@ -2,12 +2,20 @@ package pgvector_test
 
 import (
 	"encoding/json"
+	"fmt"
 	"reflect"
 	"testing"
 
 	"github.com/pgvector/pgvector-go"
 )
 
+func TestHalfVectorString(t *testing.T) {
+	vec := pgvector.NewHalfVector([]float32{1, 2, 3})
+	if fmt.Sprint(vec) != "[1,2,3]" {
+		t.Errorf("Bad marshal")
+	}
+}
+
 func TestHalfVectorMarshal(t *testing.T) {
 	vec := pgvector.NewHalfVector([]float32{1, 2, 3})
 	data, err := json.Marshal(vec)

diff --git a/sparsevec_test.go b/sparsevec_test.go
@@ -0,0 +1,15 @@
+package pgvector_test
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/pgvector/pgvector-go"
+)
+
+func TestSparseVectorString(t *testing.T) {
+	vec := pgvector.NewSparseVector([]float32{1, 0, 2, 0, 3, 0})
+	if fmt.Sprint(vec) != "{1:1,3:2,5:3}/6" {
+		t.Errorf("Bad marshal")
+	}
+}
diff --git a/vector_test.go b/vector_test.go
@@ -2,12 +2,20 @@ package pgvector_test
 
 import (
 	"encoding/json"
+	"fmt"
 	"reflect"
 	"testing"
 
 	"github.com/pgvector/pgvector-go"
 )
 
+func TestVectorString(t *testing.T) {
+	vec := pgvector.NewVector([]float32{1, 2, 3})
+	if fmt.Sprint(vec) != "[1,2,3]" {
+		t.Errorf("Bad marshal")
+	}
+}
+
 func TestVectorMarshal(t *testing.T) {
 	vec := pgvector.NewVector([]float32{1, 2, 3})
 	data, err := json.Marshal(vec)