Increase limits

rapidsai · Sep 15, 2024 · ec1a482 · ec1a482
1 parent e0d87ef
commit ec1a482
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/java/src/test/java/ai/rapids/cudf/TableTest.java b/java/src/test/java/ai/rapids/cudf/TableTest.java
@@ -8923,8 +8923,8 @@ void testParquetWriteToBufferChunked() {
     ParquetWriterOptions options = optBuilder.build();
     ParquetWriterOptions optionsNoCompress =
       optBuilder.withCompressionType(CompressionType.NONE)
-      .withRowGroupSizeRows(4)
-      .withRowGroupSizeBytes(64)
+      .withRowGroupSizeRows(10000)
+      .withRowGroupSizeBytes(10000)
       .build();
     try (Table table0 = getExpectedFileTable(columns);
          MyBufferConsumer consumer = new MyBufferConsumer()) {
@@ -9011,8 +9011,8 @@ void testParquetWriteToFileUncompressedNoStats() throws IOException {
           .withDecimalColumn("_c7", 4)
           .withDecimalColumn("_c8", 6)
           .withCompressionType(CompressionType.NONE)
-          .withRowGroupSizeRows(4)
-          .withRowGroupSizeBytes(64)
+          .withRowGroupSizeRows(10000)
+          .withRowGroupSizeBytes(10000)
           .withStatisticsFrequency(ParquetWriterOptions.StatisticsFrequency.NONE)
           .build();
       try (TableWriter writer = Table.writeParquetChunked(options, tempFile.getAbsoluteFile())) {