Added more methods to PGsparsevec

ankane · ankane · commit 6352966ab925 · 2024-06-20T00:49:36.000-07:00
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());
+    }
 }
