Method TestScan

cpp/src/arrow/dataset/test_util_internal.h:1295–1310

Shared test cases

  // Shared test cases
  void TestScan() {
    // Basic test to make sure we can scan data
    auto random_data = GetRandomData(schema({field("f64", float64())}));
    auto source = this->MakeBufferSource(random_data.get());

    this->SetDatasetSchema(random_data->schema()->fields());
    auto fragment = this->MakeFragment(*source);

    int64_t row_count = 0;
    ASSERT_OK_AND_ASSIGN(std::unique_ptr<RecordBatchReader> scanner, Scan(fragment));
    for (auto maybe_batch : *scanner) {
      ASSERT_OK_AND_ASSIGN(auto batch, maybe_batch);
      row_count += batch->num_rows();
    }
    ASSERT_EQ(row_count, GetParam().expected_rows());
  }

  // TestScanBatchSize is no longer relevant because batch size is an internal concern.
  // Consumers should only really care about batch sizing at the sink.