Added tests for Slice method

pgvector · Jun 4, 2024 · bafc9b7 · bafc9b7
1 parent 76c5936
commit bafc9b7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/halfvec_test.go b/halfvec_test.go
@@ -16,6 +16,13 @@ func TestHalfVectorString(t *testing.T) {
 	}
 }
 
+func TestHalfVectorSlice(t *testing.T) {
+	vec := pgvector.NewHalfVector([]float32{1, 2, 3})
+	if !reflect.DeepEqual(vec.Slice(), []float32{1, 2, 3}) {
+		t.Errorf("Bad slice")
+	}
+}
+
 func TestHalfVectorMarshal(t *testing.T) {
 	vec := pgvector.NewHalfVector([]float32{1, 2, 3})
 	data, err := json.Marshal(vec)

diff --git a/vector_test.go b/vector_test.go
@@ -16,6 +16,13 @@ func TestVectorString(t *testing.T) {
 	}
 }
 
+func TestVectorSlice(t *testing.T) {
+	vec := pgvector.NewVector([]float32{1, 2, 3})
+	if !reflect.DeepEqual(vec.Slice(), []float32{1, 2, 3}) {
+		t.Errorf("Bad slice")
+	}
+}
+
 func TestVectorMarshal(t *testing.T) {
 	vec := pgvector.NewVector([]float32{1, 2, 3})
 	data, err := json.Marshal(vec)