fix: reviews v2

vibhatha · Feb 1, 2024 · d09533c · d09533c
1 parent 3bce128
commit d09533c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/...sion/src/test/java/org/apache/arrow/compression/TestArrowReaderWriterWithCompression.java b/...sion/src/test/java/org/apache/arrow/compression/TestArrowReaderWriterWithCompression.java
@@ -54,19 +54,21 @@
 import org.apache.arrow.vector.util.ByteArrayReadableSeekableByteChannel;
 import org.junit.After;
 import org.junit.Assert;
-import org.junit.Before;
-import org.junit.Test;
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Disabled;
+import org.junit.jupiter.api.Test;
 
 public class TestArrowReaderWriterWithCompression {
 
   private BufferAllocator allocator;
   private ByteArrayOutputStream out;
   private VectorSchemaRoot root;
 
-  @Before
+  @BeforeEach
   public void setup() {
-    allocator = new RootAllocator(Integer.MAX_VALUE);
+    if (allocator == null) {
+      allocator = new RootAllocator(Integer.MAX_VALUE);
+    }
     out = new ByteArrayOutputStream();
     root = null;
   }