Fixed an issue with scaling on parquet read tests

deephaven · Oct 10, 2023 · b713313 · b713313
1 parent 2c33f5a
commit b713313
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/it/java/io/deephaven/benchmark/tests/standard/parquet/ParquetCodecTest.java b/src/it/java/io/deephaven/benchmark/tests/standard/parquet/ParquetCodecTest.java
@@ -14,11 +14,15 @@
 public class ParquetCodecTest {
     final ParquetTestRunner runner = new ParquetTestRunner(this);
     final String[] usedColumns = {"str10K", "long10K", "int10K", "short10K", "bigDec10K", "array1K", "vector1K"};
+
+    @BeforeEach
+    public void setup() {
+        runner.setScaleFactors(5, 1);
+    }
 
     @Test
     @Order(1)
     public void writeMultiColSnappy() {
-        runner.setScaleFactors(5, 1);
         runner.runWriteTest("ParquetWrite- Snappy Multi Col -Static", "SNAPPY", usedColumns);
     }
 
@@ -31,7 +35,6 @@ public void readMultiColSnappy() {
     @Test
     @Order(3)
     public void writeMultiColZstd() {
-        runner.setScaleFactors(5, 1);
         runner.runWriteTest("ParquetWrite- Zstd Multi Col -Static", "ZSTD", usedColumns);
     }
 
@@ -44,7 +47,6 @@ public void readMultiColZstd() {
     @Test
     @Order(5)
     public void writeMultiColLzo() {
-        runner.setScaleFactors(5, 1);
         runner.runWriteTest("ParquetWrite- Lzo Multi Col -Static", "LZO", usedColumns);
     }
 
@@ -57,7 +59,6 @@ public void readMultiColLzo() {
     @Test
     @Order(7)
     public void writeMultiColLz4Raw() {
-        runner.setScaleFactors(5, 1);
         runner.runWriteTest("ParquetWrite- Lz4Raw Multi Col -Static", "LZ4_RAW", usedColumns);
     }
 
@@ -70,7 +71,6 @@ public void readMultiColLz4Raw() {
     @Test
     @Order(9)
     public void writeMultiColGzip() {
-        runner.setScaleFactors(5, 1);
         runner.runWriteTest("ParquetWrite- Gzip Multi Col -Static", "GZIP", usedColumns);
     }
 
@@ -83,7 +83,6 @@ public void readMultiColGzip() {
     @Test
     @Order(11)
     public void writeMultiColNone() {
-        runner.setScaleFactors(5, 1);
         runner.runWriteTest("ParquetWrite- No Codec Multi Col -Static", "NONE", usedColumns);
     }