MINOR: [JAVA] Add unit test for float16 into cdata interface module (a…

…pache#40185) Add unit test for float16 into CData Interface module. Authored-by: david dali susanibar arce <davi.sarces@gmail.com> Signed-off-by: David Li <li.davidm96@gmail.com>
vibhatha · Feb 21, 2024 · e198f30 · e198f30
1 parent 884a109
commit e198f30
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/java/c/src/test/java/org/apache/arrow/c/RoundtripTest.java b/java/c/src/test/java/org/apache/arrow/c/RoundtripTest.java
@@ -48,6 +48,7 @@
 import org.apache.arrow.vector.ExtensionTypeVector;
 import org.apache.arrow.vector.FieldVector;
 import org.apache.arrow.vector.FixedSizeBinaryVector;
+import org.apache.arrow.vector.Float2Vector;
 import org.apache.arrow.vector.Float4Vector;
 import org.apache.arrow.vector.Float8Vector;
 import org.apache.arrow.vector.IntVector;
@@ -297,6 +298,14 @@ public void testFixedSizeBinaryVector() {
     }
   }
 
+  @Test
+  public void testFloat2Vector() {
+    try (final Float2Vector vector = new Float2Vector("v", allocator)) {
+      setVector(vector, 0.1f, 0.2f, 0.3f, null);
+      assertTrue(roundtrip(vector, Float2Vector.class));
+    }
+  }
+
   @Test
   public void testFloat4Vector() {
     try (final Float4Vector vector = new Float4Vector("v", allocator)) {

diff --git a/java/vector/src/test/java/org/apache/arrow/vector/testing/ValueVectorDataPopulator.java b/java/vector/src/test/java/org/apache/arrow/vector/testing/ValueVectorDataPopulator.java
@@ -34,6 +34,7 @@
 import org.apache.arrow.vector.DecimalVector;
 import org.apache.arrow.vector.DurationVector;
 import org.apache.arrow.vector.FixedSizeBinaryVector;
+import org.apache.arrow.vector.Float2Vector;
 import org.apache.arrow.vector.Float4Vector;
 import org.apache.arrow.vector.Float8Vector;
 import org.apache.arrow.vector.IntVector;
@@ -207,6 +208,20 @@ public static void setVector(FixedSizeBinaryVector vector, byte[]... values) {
     vector.setValueCount(length);
   }
 
+  /**
+   * Populate values for Float2Vector.
+   */
+  public static void setVector(Float2Vector vector, Float... values) {
+    final int length = values.length;
+    vector.allocateNew(length);
+    for (int i = 0; i < length; i++) {
+      if (values[i] != null) {
+        vector.setWithPossibleTruncate(i, values[i]);
+      }
+    }
+    vector.setValueCount(length);
+  }
+
   /**
    * Populate values for Float4Vector.
    */