MCPcopy Create free account
hub / github.com/apache/arrow / UnequalNumRows

Method UnequalNumRows

cpp/src/parquet/file_serialize_test.cc:183–204  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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) {

Callers 1

TYPED_TESTFunction · 0.80

Calls 9

CreateOutputStreamFunction · 0.85
BuilderFunction · 0.70
buildMethod · 0.45
AppendRowGroupMethod · 0.45
GenerateDataMethod · 0.45
NextColumnMethod · 0.45
WriteBatchMethod · 0.45
dataMethod · 0.45
CloseMethod · 0.45

Tested by

no test coverage detected