remove debug printlines in test

Author: ruolin59

coral-schema/src/main/java/com/linkedin/coral/schema/avro/MergeHiveSchemaWithAvro.java

@@ -177,19 +177,13 @@ public Schema mapKeyPartner(Schema partner) {
     @Override
     public Schema mapValuePartner(Schema partner) {
       Schema schema = SchemaUtilities.extractIfOption(partner);
-      if (schema.getType() != Schema.Type.MAP) {
-        return null;
-      }
-      return schema.getValueType();
+      return (schema.getType() == Schema.Type.MAP) ? schema.getValueType() : null;
     }
 
     @Override
     public Schema listElementPartner(Schema partner) {
       Schema schema = SchemaUtilities.extractIfOption(partner);
-      if (schema.getType() != Schema.Type.ARRAY) {
-        return null;
-      }
-      return schema.getElementType();
+      return (schema.getType() == Schema.Type.ARRAY) ? schema.getElementType() : null;
     }
 
     @Override

coral-schema/src/test/java/com/linkedin/coral/schema/avro/MergeHiveSchemaWithAvroTests.java

@@ -273,26 +273,18 @@ public void shouldHandleSingleElementUnionsInAllTypes() {
 
     // Construct Avro schema with single-element unions for primitives, array items, and map values
     Schema avro = struct("test_complex_array_table", "doc-test", "com.example.test", optional("id", Schema.Type.LONG),
-        field("name", singleElementUnion(Schema.create(Schema.Type.STRING)), null, "unknown", null),
-        field("active", singleElementUnion(Schema.create(Schema.Type.BOOLEAN)), null, false, null),
-        optional("items", array(singleElementUnion(seu_arrayItemConfigSchema))),
-        optional("metadata", map(singleElementUnion(seu_mapValueMetadataSchema))),
-        optional("tags", array(singleElementUnion(Schema.create(Schema.Type.STRING)))));
+        field("name", SchemaUtilities.wrapInSingleElementUnion(Schema.create(Schema.Type.STRING)), null, "unknown", null),
+        field("active", SchemaUtilities.wrapInSingleElementUnion(Schema.create(Schema.Type.BOOLEAN)), null, false, null),
+        optional("items", array(SchemaUtilities.wrapInSingleElementUnion(seu_arrayItemConfigSchema))),
+        optional("metadata", map(SchemaUtilities.wrapInSingleElementUnion(seu_mapValueMetadataSchema))),
+        optional("tags", array(SchemaUtilities.wrapInSingleElementUnion(Schema.create(Schema.Type.STRING)))));
 
     // Expected schema after merge: single-element unions should be preserved
     // The structure of the Avro schema is maintained, including single-element unions
     // Expected output should match the input avro schema
     Schema expected = avro;
 
     Schema actual = merge(hive, avro);
-    
-    System.out.println("\n=== INPUT AVRO SCHEMA ===");
-    System.out.println(avro.toString(true));
-    System.out.println("\n=== EXPECTED OUTPUT SCHEMA ===");
-    System.out.println(expected.toString(true));
-    System.out.println("\n=== ACTUAL OUTPUT SCHEMA ===");
-    System.out.println(actual.toString(true));
-    System.out.println("\n=== END ===\n");
 
     assertSchema(expected, actual);
   }
@@ -322,23 +314,15 @@ public void shouldHandleSingleElementUnionsWithHiveUnionType() {
     // Also includes a regular union for the status field (to test Hive uniontype compatibility)
     Schema avro = struct("test_union_compat", "doc-test", "com.example.test", optional("id", Schema.Type.LONG),
         required("status", union(Schema.Type.NULL, Schema.Type.STRING, Schema.Type.INT)),
-        optional("items", array(singleElementUnion(seu_arrayItemSchema))),
-        optional("metadata", map(singleElementUnion(seu_mapValueMetadataSchema))));
+        optional("items", array(SchemaUtilities.wrapInSingleElementUnion(seu_arrayItemSchema))),
+        optional("metadata", map(SchemaUtilities.wrapInSingleElementUnion(seu_mapValueMetadataSchema))));
 
     // Expected schema: single-element unions should be preserved, regular union preserved
     // The Avro schema structure is maintained
     // Expected output should match the input avro schema
     Schema expected = avro;
 
     Schema actual = merge(hive, avro);
-    
-    System.out.println("\n=== INPUT AVRO SCHEMA (HiveUnionType test) ===");
-    System.out.println(avro.toString(true));
-    System.out.println("\n=== EXPECTED OUTPUT SCHEMA (HiveUnionType test) ===");
-    System.out.println(expected.toString(true));
-    System.out.println("\n=== ACTUAL OUTPUT SCHEMA (HiveUnionType test) ===");
-    System.out.println(actual.toString(true));
-    System.out.println("\n=== END ===\n");
 
     assertSchema(expected, actual);
   }
@@ -353,10 +337,6 @@ private Schema union(Schema.Type... types) {
     return Schema.createUnion(Arrays.stream(types).map(Schema::create).collect(Collectors.toList()));
   }
 
-  private Schema singleElementUnion(Schema schema) {
-    return Schema.createUnion(Arrays.asList(schema));
-  }
-
   private void assertSchema(Schema expected, Schema actual) {
     assertEquals(actual.toString(true), expected.toString(true));
   }