| 268 | } |
| 269 | |
| 270 | void ZeroRowsRowGroup() { |
| 271 | auto sink = CreateOutputStream(); |
| 272 | auto gnode = std::static_pointer_cast<GroupNode>(this->node_); |
| 273 | |
| 274 | std::shared_ptr<WriterProperties> props = WriterProperties::Builder().build(); |
| 275 | |
| 276 | auto file_writer = ParquetFileWriter::Open(sink, gnode, props); |
| 277 | |
| 278 | RowGroupWriter* row_group_writer; |
| 279 | |
| 280 | row_group_writer = file_writer->AppendRowGroup(); |
| 281 | for (int col = 0; col < num_columns_; ++col) { |
| 282 | auto column_writer = |
| 283 | static_cast<TypedColumnWriter<TestType>*>(row_group_writer->NextColumn()); |
| 284 | column_writer->Close(); |
| 285 | } |
| 286 | row_group_writer->Close(); |
| 287 | |
| 288 | row_group_writer = file_writer->AppendBufferedRowGroup(); |
| 289 | for (int col = 0; col < num_columns_; ++col) { |
| 290 | auto column_writer = |
| 291 | static_cast<TypedColumnWriter<TestType>*>(row_group_writer->column(col)); |
| 292 | column_writer->Close(); |
| 293 | } |
| 294 | row_group_writer->Close(); |
| 295 | |
| 296 | file_writer->Close(); |
| 297 | } |
| 298 | }; |
| 299 | |
| 300 | typedef ::testing::Types<Int32Type, Int64Type, Int96Type, FloatType, DoubleType, |
no test coverage detected