add test to TestTypeBuilders

Author: zhangfengcdt

parquet-column/src/test/java/org/apache/parquet/schema/TestTypeBuilders.java

@@ -55,6 +55,7 @@
 import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.Callable;
+import org.apache.parquet.format.EdgeInterpolationAlgorithm;
 import org.apache.parquet.schema.PrimitiveType.PrimitiveTypeName;
 import org.apache.parquet.schema.Type.Repetition;
 import org.junit.Assert;
@@ -1101,7 +1102,7 @@ public void testMapWithPreBuiltKeyAndValueTypes() {
 
     GroupType map = new GroupType(
         REQUIRED, "myMap", OriginalType.MAP, new GroupType(REPEATED, "key_value", new Type[] {keyType, valueType
-        }));
+    }));
     MessageType expected = new MessageType("mapParent", map);
 
     GroupType actual = Types.buildMessage()
@@ -1477,6 +1478,105 @@ public void testDecimalLogicalTypeWithDeprecatedPrecisionMismatch() {
         .named("aDecimal");
   }
 
+  @Test
+  public void testGeometryLogicalType() {
+    // Test with default CRS
+    PrimitiveType defaultCrsExpected = new PrimitiveType(
+        REQUIRED, BINARY, "aGeometry", LogicalTypeAnnotation.geometryType("OGC:CRS84"));
+    PrimitiveType defaultCrsActual = Types.required(BINARY)
+        .as(LogicalTypeAnnotation.geometryType("OGC:CRS84"))
+        .named("aGeometry");
+    Assert.assertEquals(defaultCrsExpected, defaultCrsActual);
+
+    // Test with custom CRS
+    PrimitiveType customCrsExpected = new PrimitiveType(
+        REQUIRED, BINARY, "aGeometry", LogicalTypeAnnotation.geometryType("EPSG:4326"));
+    PrimitiveType customCrsActual = Types.required(BINARY)
+        .as(LogicalTypeAnnotation.geometryType("EPSG:4326"))
+        .named("aGeometry");
+    Assert.assertEquals(customCrsExpected, customCrsActual);
+
+    // Test with optional repetition
+    PrimitiveType optionalGeometryExpected = new PrimitiveType(
+        OPTIONAL, BINARY, "aGeometry", LogicalTypeAnnotation.geometryType("OGC:CRS84"));
+    PrimitiveType optionalGeometryActual = Types.optional(BINARY)
+        .as(LogicalTypeAnnotation.geometryType("OGC:CRS84"))
+        .named("aGeometry");
+    Assert.assertEquals(optionalGeometryExpected, optionalGeometryActual);
+  }
+
+  @Test
+  public void testGeographyLogicalType() {
+    // Test with default CRS and no edge algorithm
+    PrimitiveType defaultCrsExpected = new PrimitiveType(
+        REQUIRED, BINARY, "aGeography", LogicalTypeAnnotation.geographyType("OGC:CRS84", null));
+    PrimitiveType defaultCrsActual = Types.required(BINARY)
+        .as(LogicalTypeAnnotation.geographyType("OGC:CRS84", null))
+        .named("aGeography");
+    Assert.assertEquals(defaultCrsExpected, defaultCrsActual);
+
+    // Test with custom CRS and no edge algorithm
+    PrimitiveType customCrsExpected = new PrimitiveType(
+        REQUIRED, BINARY, "aGeography", LogicalTypeAnnotation.geographyType("EPSG:4326", null));
+    PrimitiveType customCrsActual = Types.required(BINARY)
+        .as(LogicalTypeAnnotation.geographyType("EPSG:4326", null))
+        .named("aGeography");
+    Assert.assertEquals(customCrsExpected, customCrsActual);
+
+    // Test with custom CRS and edge algorithm
+    EdgeInterpolationAlgorithm greatCircle = EdgeInterpolationAlgorithm.SPHERICAL;
+    PrimitiveType customCrsWithEdgeAlgorithmExpected = new PrimitiveType(
+        REQUIRED, BINARY, "aGeography", LogicalTypeAnnotation.geographyType("EPSG:4326", null));
+    PrimitiveType customCrsWithEdgeAlgorithmActual = Types.required(BINARY)
+        .as(LogicalTypeAnnotation.geographyType("EPSG:4326", null))
+        .named("aGeography");
+    Assert.assertEquals(customCrsWithEdgeAlgorithmExpected, customCrsWithEdgeAlgorithmActual);
+
+    // Test with optional repetition
+    PrimitiveType optionalGeographyExpected = new PrimitiveType(
+        OPTIONAL, BINARY, "aGeography", LogicalTypeAnnotation.geographyType("OGC:CRS84", null));
+    PrimitiveType optionalGeographyActual = Types.optional(BINARY)
+        .as(LogicalTypeAnnotation.geographyType("OGC:CRS84", null))
+        .named("aGeography");
+    Assert.assertEquals(optionalGeographyExpected, optionalGeographyActual);
+  }
+
+  @Test
+  public void testGeographyLogicalTypeWithoutEdgeInterpolationAlgorithm() {
+    // Test with default CRS and no edge algorithm
+    PrimitiveType defaultCrsExpected = new PrimitiveType(
+        REQUIRED, BINARY, "aGeography", LogicalTypeAnnotation.geographyType());
+    PrimitiveType defaultCrsActual = Types.required(BINARY)
+        .as(LogicalTypeAnnotation.geographyType())
+        .named("aGeography");
+    Assert.assertEquals(defaultCrsExpected, defaultCrsActual);
+
+    // Test with custom CRS and no edge algorithm
+    PrimitiveType customCrsExpected = new PrimitiveType(
+        REQUIRED, BINARY, "aGeography", LogicalTypeAnnotation.geographyType("EPSG:4326", null));
+    PrimitiveType customCrsActual = Types.required(BINARY)
+        .as(LogicalTypeAnnotation.geographyType("EPSG:4326", null))
+        .named("aGeography");
+    Assert.assertEquals(customCrsExpected, customCrsActual);
+
+    // Test with custom CRS and edge algorithm
+    PrimitiveType customCrsWithEdgeAlgorithmExpected = new PrimitiveType(
+        REQUIRED, BINARY, "aGeography",
+        LogicalTypeAnnotation.geographyType("EPSG:4326", null));
+    PrimitiveType customCrsWithEdgeAlgorithmActual = Types.required(BINARY)
+        .as(LogicalTypeAnnotation.geographyType("EPSG:4326", null))
+        .named("aGeography");
+    Assert.assertEquals(customCrsWithEdgeAlgorithmExpected, customCrsWithEdgeAlgorithmActual);
+
+    // Test with optional repetition
+    PrimitiveType optionalGeographyExpected = new PrimitiveType(
+        OPTIONAL, BINARY, "aGeography", LogicalTypeAnnotation.geographyType());
+    PrimitiveType optionalGeographyActual = Types.optional(BINARY)
+        .as(LogicalTypeAnnotation.geographyType())
+        .named("aGeography");
+    Assert.assertEquals(optionalGeographyExpected, optionalGeographyActual);
+  }
+
   /**
    * A convenience method to avoid a large number of @Test(expected=...) tests
    *