| 179 | // TODO: Assert on definition and repetition levels |
| 180 | template <typename DType, typename ValueType = typename DType::c_type> |
| 181 | void AssertColumnValues(std::shared_ptr<TypedColumnReader<DType>> col, int64_t batch_size, |
| 182 | int64_t expected_levels_read, |
| 183 | const std::vector<ValueType>& expected_values, |
| 184 | int64_t expected_values_read) { |
| 185 | std::vector<ValueType> values(batch_size); |
| 186 | int64_t values_read; |
| 187 | |
| 188 | auto levels_read = |
| 189 | col->ReadBatch(batch_size, nullptr, nullptr, values.data(), &values_read); |
| 190 | ASSERT_EQ(expected_levels_read, levels_read); |
| 191 | ASSERT_EQ(expected_values_read, values_read); |
| 192 | AssertColumnValuesEqual(col->descr(), expected_values, values); |
| 193 | } |
| 194 | |
| 195 | template <typename DType, typename ValueType = typename DType::c_type> |
| 196 | void AssertColumnValuesEqual(std::shared_ptr<TypedColumnReader<DType>> left_col, |
no test coverage detected