| 147 | |
| 148 | template <typename DType, typename ValueType = typename DType::c_type> |
| 149 | std::vector<ValueType> ReadColumnValues(ParquetFileReader* file_reader, int row_group, |
| 150 | int column, int64_t expected_values_read) { |
| 151 | auto column_reader = checked_pointer_cast<TypedColumnReader<DType>>( |
| 152 | file_reader->RowGroup(row_group)->Column(column)); |
| 153 | std::vector<ValueType> values(expected_values_read); |
| 154 | int64_t values_read; |
| 155 | auto levels_read = column_reader->ReadBatch(expected_values_read, nullptr, nullptr, |
| 156 | values.data(), &values_read); |
| 157 | EXPECT_EQ(expected_values_read, levels_read); |
| 158 | EXPECT_EQ(expected_values_read, values_read); |
| 159 | return values; |
| 160 | } |
| 161 | |
| 162 | template <typename ValueType> |
| 163 | void AssertColumnValuesEqual(const ColumnDescriptor* descr, |