Add tests

Author: JingsongLi

java/vortex-jni/src/test/java/dev/vortex/api/DTypeTest.java

@@ -0,0 +1,81 @@
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: Copyright the Vortex contributors
+
+package dev.vortex.api;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+import org.junit.jupiter.api.Test;
+
+public final class DTypeTest {
+
+    @Test
+    public void testNewFixedSizeListNonNullable() {
+        try (var elementType = DType.newInt(false);
+                var fslType = DType.newFixedSizeList(elementType, 3, false)) {
+            assertEquals(DType.Variant.FIXED_SIZE_LIST, fslType.getVariant());
+            assertFalse(fslType.isNullable());
+            assertEquals(3, fslType.getFixedSizeListSize());
+
+            var innerType = fslType.getElementType();
+            assertEquals(DType.Variant.PRIMITIVE_I32, innerType.getVariant());
+        }
+    }
+
+    @Test
+    public void testNewFixedSizeListNullable() {
+        try (var elementType = DType.newUtf8(true);
+                var fslType = DType.newFixedSizeList(elementType, 5, true)) {
+            assertEquals(DType.Variant.FIXED_SIZE_LIST, fslType.getVariant());
+            assertTrue(fslType.isNullable());
+            assertEquals(5, fslType.getFixedSizeListSize());
+
+            var innerType = fslType.getElementType();
+            assertEquals(DType.Variant.UTF8, innerType.getVariant());
+        }
+    }
+
+    @Test
+    public void testNewListGetElementType() {
+        try (var elementType = DType.newDouble(false);
+                var listType = DType.newList(elementType, false)) {
+            assertEquals(DType.Variant.LIST, listType.getVariant());
+
+            var innerType = listType.getElementType();
+            assertEquals(DType.Variant.PRIMITIVE_F64, innerType.getVariant());
+        }
+    }
+
+    @Test
+    public void testNestedFixedSizeList() {
+        try (var innerElement = DType.newLong(false);
+                var innerFsl = DType.newFixedSizeList(innerElement, 2, false);
+                var outerFsl = DType.newFixedSizeList(innerFsl, 4, true)) {
+            assertEquals(DType.Variant.FIXED_SIZE_LIST, outerFsl.getVariant());
+            assertTrue(outerFsl.isNullable());
+            assertEquals(4, outerFsl.getFixedSizeListSize());
+
+            var inner = outerFsl.getElementType();
+            assertEquals(DType.Variant.FIXED_SIZE_LIST, inner.getVariant());
+        }
+    }
+
+    @Test
+    public void testFixedSizeListInStruct() {
+        try (var elementType = DType.newFloat(false);
+                var fslType = DType.newFixedSizeList(elementType, 3, false);
+                var structType = DType.newStruct(
+                        new String[] {"id", "embedding"},
+                        new DType[] {DType.newInt(false), fslType},
+                        false)) {
+            assertEquals(DType.Variant.STRUCT, structType.getVariant());
+
+            var fieldTypes = structType.getFieldTypes();
+            assertEquals(2, fieldTypes.size());
+
+            try (var embeddingType = fieldTypes.get(1)) {
+                assertEquals(DType.Variant.FIXED_SIZE_LIST, embeddingType.getVariant());
+            }
+        }
+    }
+}

java/vortex-spark/src/test/java/dev/vortex/spark/SparkTypesTest.java

@@ -0,0 +1,46 @@
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: Copyright the Vortex contributors
+
+package dev.vortex.spark;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+import dev.vortex.api.DType;
+import dev.vortex.jni.NativeLoader;
+import org.apache.spark.sql.types.ArrayType;
+import org.apache.spark.sql.types.DataTypes;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.DisplayName;
+import org.junit.jupiter.api.Test;
+
+public final class SparkTypesTest {
+
+    @BeforeAll
+    public static void loadLibrary() {
+        NativeLoader.loadJni();
+    }
+
+    @Test
+    @DisplayName("toDataType should convert FIXED_SIZE_LIST to Spark ArrayType")
+    public void testFixedSizeListToDataType() {
+        try (var elementType = DType.newInt(false);
+                var fslType = DType.newFixedSizeList(elementType, 3, true)) {
+            var sparkType = SparkTypes.toDataType(fslType);
+            assertInstanceOf(ArrayType.class, sparkType);
+            ArrayType arrayType = (ArrayType) sparkType;
+            assertEquals(DataTypes.IntegerType, arrayType.elementType());
+        }
+    }
+
+    @Test
+    @DisplayName("toDataType should convert LIST to Spark ArrayType")
+    public void testListToDataType() {
+        try (var elementType = DType.newDouble(false);
+                var listType = DType.newList(elementType, true)) {
+            var sparkType = SparkTypes.toDataType(listType);
+            assertInstanceOf(ArrayType.class, sparkType);
+            ArrayType arrayType = (ArrayType) sparkType;
+            assertEquals(DataTypes.DoubleType, arrayType.elementType());
+        }
+    }
+}