From 138d5f4d68b7e3343959d5aea4a252456eb6ded5 Mon Sep 17 00:00:00 2001 From: seidl Date: Fri, 3 Feb 2023 13:28:07 -0800 Subject: [PATCH] fix structs of lists test --- cpp/tests/io/parquet_chunked_reader_test.cpp | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/cpp/tests/io/parquet_chunked_reader_test.cpp b/cpp/tests/io/parquet_chunked_reader_test.cpp index 0cecf62cc9d..18349b6a1dc 100644 --- a/cpp/tests/io/parquet_chunked_reader_test.cpp +++ b/cpp/tests/io/parquet_chunked_reader_test.cpp @@ -62,8 +62,9 @@ using int32s_lists_col = cudf::test::lists_column_wrapper; auto write_file(std::vector>& input_columns, std::string const& filename, bool nullable, - std::size_t max_page_size_bytes = cudf::io::default_max_page_size_bytes, - std::size_t max_page_size_rows = cudf::io::default_max_page_size_rows) + std::size_t max_page_size_bytes = cudf::io::default_max_page_size_bytes, + std::size_t max_page_size_rows = cudf::io::default_max_page_size_rows, + cudf::size_type max_page_fragment_size = cudf::io::default_max_page_fragment_size) { // Just shift nulls of the next column by one position to avoid having all nulls in the same // table rows. @@ -93,6 +94,7 @@ auto write_file(std::vector>& input_columns, cudf::io::parquet_writer_options::builder(cudf::io::sink_info{filepath}, *input_table) .max_page_size_bytes(max_page_size_bytes) .max_page_size_rows(max_page_size_rows) + .max_page_fragment_size(max_page_fragment_size) .build(); cudf::io::write_parquet(write_opts); @@ -637,7 +639,8 @@ TEST_F(ParquetChunkedReaderTest, TestChunkedReadWithStructsOfLists) "chunked_read_with_structs_of_lists", nullable, 512 * 1024, // 512KB per page - 20000 // 20k rows per page + 20000, // 20k rows per page + 5001 // non-default value to prevent variable fragments ); };