Added more methods to PGsparsevec

pgvector · Jun 20, 2024 · 6352966 · 6352966
1 parent c9f2996
commit 6352966
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 0 deletions.
diff --git a/src/main/java/com/pgvector/PGsparsevec.java b/src/main/java/com/pgvector/PGsparsevec.java
@@ -220,6 +220,33 @@ public float[] toArray() {
         return vec;
     }
 
+    /**
+     * Returns the number of dimensions
+     *
+     * @return the number of dimensions
+     */
+    public int getDimensions() {
+        return dimensions;
+    }
+
+    /**
+     * Returns the non-zero indices
+     *
+     * @return the non-zero indices
+     */
+    public int[] getIndices() {
+        return indices;
+    }
+
+    /**
+     * Returns the non-zero values
+     *
+     * @return the non-zero values
+     */
+    public float[] getValues() {
+        return values;
+    }
+
     /**
      * Registers the sparsevec type
      *

diff --git a/src/test/java/com/pgvector/PGsparsevecTest.java b/src/test/java/com/pgvector/PGsparsevecTest.java
@@ -40,4 +40,22 @@ void testGetValue() {
         PGsparsevec vec = new PGsparsevec(new float[] {1, 0, 2, 0, 3, 0});
         assertEquals("{1:1.0,3:2.0,5:3.0}/6", vec.getValue());
     }
+
+    @Test
+    void testGetDimensions() {
+        PGsparsevec vec = new PGsparsevec(new float[] {1, 0, 2, 0, 3, 0});
+        assertEquals(6, vec.getDimensions());
+    }
+
+    @Test
+    void testGetIndices() {
+        PGsparsevec vec = new PGsparsevec(new float[] {1, 0, 2, 0, 3, 0});
+        assertArrayEquals(new int[] {0, 2, 4}, vec.getIndices());
+    }
+
+    @Test
+    void testGetValues() {
+        PGsparsevec vec = new PGsparsevec(new float[] {1, 0, 2, 0, 3, 0});
+        assertArrayEquals(new float[] {1, 2, 3}, vec.getValues());
+    }
 }