Minor refactoring to RoaringBitmapArraySuite.

GitOrigin-RevId: eb77c958731b273964277b6dc01f033df573bd25
the2nd07 · Nov 29, 2022 · 80b1224 · 80b1224
1 parent b3ff96c
commit 80b1224
Showing 1 changed file with 21 additions and 18 deletions.
diff --git a/core/src/test/scala/org/apache/spark/sql/delta/deletionvectors/RoaringBitmapArraySuite.scala b/core/src/test/scala/org/apache/spark/sql/delta/deletionvectors/RoaringBitmapArraySuite.scala
@@ -122,13 +122,14 @@ class RoaringBitmapArraySuite extends SparkFunSuite {
     bitmap
   }
 
-  for (serializationFormat <- RoaringBitmapArrayFormat.values)
-  test(s"serialization - $serializationFormat") {
-    checkSerializeDeserialize(RoaringBitmapArray(), serializationFormat)
-    checkSerializeDeserialize(RoaringBitmapArray(1L), serializationFormat)
-    checkSerializeDeserialize(RoaringBitmapArray(BITMAP2_NUMBER), serializationFormat)
-    checkSerializeDeserialize(RoaringBitmapArray(1L, BITMAP2_NUMBER), serializationFormat)
-    checkSerializeDeserialize(allContainerTypesBitmap, serializationFormat)
+  for (serializationFormat <- RoaringBitmapArrayFormat.values) {
+    test(s"serialization - $serializationFormat") {
+      checkSerializeDeserialize(RoaringBitmapArray(), serializationFormat)
+      checkSerializeDeserialize(RoaringBitmapArray(1L), serializationFormat)
+      checkSerializeDeserialize(RoaringBitmapArray(BITMAP2_NUMBER), serializationFormat)
+      checkSerializeDeserialize(RoaringBitmapArray(1L, BITMAP2_NUMBER), serializationFormat)
+      checkSerializeDeserialize(allContainerTypesBitmap, serializationFormat)
+    }
   }
 
   private def checkSerializeDeserialize(
@@ -143,19 +144,21 @@ class RoaringBitmapArraySuite extends SparkFunSuite {
     assert(input === output)
   }
 
-  for (serializationFormat <- RoaringBitmapArrayFormat.values)
-  test(s"serialization and deserialization with big endian buffers throws - $serializationFormat") {
-    val roaringBitmapArray = RoaringBitmapArray(1L)
-    val bigEndianBuffer = ByteBuffer
-      .allocate(roaringBitmapArray.serializedSizeInBytes(serializationFormat).toInt)
-      .order(ByteOrder.BIG_ENDIAN)
+  for (serializationFormat <- RoaringBitmapArrayFormat.values) {
+    test(
+      s"serialization and deserialization with big endian buffers throws - $serializationFormat") {
+      val roaringBitmapArray = RoaringBitmapArray(1L)
+      val bigEndianBuffer = ByteBuffer
+        .allocate(roaringBitmapArray.serializedSizeInBytes(serializationFormat).toInt)
+        .order(ByteOrder.BIG_ENDIAN)
 
-    assertThrows[IllegalArgumentException] {
-      roaringBitmapArray.serialize(bigEndianBuffer, serializationFormat)
-    }
+      assertThrows[IllegalArgumentException] {
+        roaringBitmapArray.serialize(bigEndianBuffer, serializationFormat)
+      }
 
-    assertThrows[IllegalArgumentException] {
-      roaringBitmapArray.deserialize(bigEndianBuffer)
+      assertThrows[IllegalArgumentException] {
+        roaringBitmapArray.deserialize(bigEndianBuffer)
+      }
     }
   }