| 181 | } |
| 182 | |
| 183 | void UnequalNumRows(int64_t max_rows, const std::vector<int64_t> rows_per_column) { |
| 184 | auto sink = CreateOutputStream(); |
| 185 | auto gnode = std::static_pointer_cast<GroupNode>(this->node_); |
| 186 | |
| 187 | std::shared_ptr<WriterProperties> props = WriterProperties::Builder().build(); |
| 188 | |
| 189 | auto file_writer = ParquetFileWriter::Open(sink, gnode, props); |
| 190 | |
| 191 | RowGroupWriter* row_group_writer; |
| 192 | row_group_writer = file_writer->AppendRowGroup(); |
| 193 | |
| 194 | this->GenerateData(max_rows); |
| 195 | for (int col = 0; col < num_columns_; ++col) { |
| 196 | auto column_writer = |
| 197 | static_cast<TypedColumnWriter<TestType>*>(row_group_writer->NextColumn()); |
| 198 | column_writer->WriteBatch(rows_per_column[col], this->def_levels_.data(), nullptr, |
| 199 | this->values_ptr_); |
| 200 | column_writer->Close(); |
| 201 | } |
| 202 | row_group_writer->Close(); |
| 203 | file_writer->Close(); |
| 204 | } |
| 205 | |
| 206 | void UnequalNumRowsBuffered(int64_t max_rows, |
| 207 | const std::vector<int64_t> rows_per_column) { |
no test coverage detected