Method CheckDecodeArrow

cpp/src/parquet/encoding_test.cc:792–824

  }

  void CheckDecodeArrow(Encoding::type encoding) {
    for (double np : null_probabilities_) {
      for (double true_prob : true_probabilities_) {
        InitTestCase(encoding, np, true_prob);
        for (int read_batch_size : this->read_batch_sizes_) {
          ResetTheDecoder();

          int num_values_left = kNumValues;
          ::arrow::BooleanBuilder acc;
          int actual_num_not_null_values = 0;
          while (num_values_left > 0) {
            int batch_size = std::min(num_values_left, read_batch_size);
            ASSERT_NE(0, batch_size);
            // Counting nulls
            int64_t batch_null_count = 0;
            if (null_count_ != 0) {
              batch_null_count =
                  batch_size - ::arrow::internal::CountSetBits(
                                   valid_bits_, kNumValues - num_values_left, batch_size);
            }
            int batch_num_values =
                decoder_->DecodeArrow(batch_size, static_cast<int>(batch_null_count),
                                      valid_bits_, kNumValues - num_values_left, &acc);
            actual_num_not_null_values += batch_num_values;
            num_values_left -= batch_size;
          }
          std::shared_ptr<::arrow::Array> chunk;
          ASSERT_OK(acc.Finish(&chunk));
          CheckDense(actual_num_not_null_values, *chunk);
        }
      }
    }
  }

  void CheckDecodeArrowNonNull(Encoding::type encoding) {
    // NonNull skips tests for null_prob != 0.