Function VerifySupplier

cpp/src/arrow/acero/tpch_node_test.cc:346–371

}

void VerifySupplier(const std::vector<ExecBatch>& batches,
                    double scale_factor = kDefaultScaleFactor) {
  int64_t kExpectedRows = static_cast<int64_t>(10000 * scale_factor);
  int64_t num_rows = 0;

  std::unordered_set<int32_t> seen_suppkey;
  int good_count = 0;
  int bad_count = 0;
  for (auto& batch : batches) {
    ValidateBatch(batch);
    VerifyUniqueKey(&seen_suppkey, batch[0],
                    /*min=*/1,
                    /*max=*/static_cast<int32_t>(kExpectedRows));
    VerifyStringAndNumber_FixedWidth(batch[1], batch[0], /*byte_width=*/25, "Supplie#r");
    VerifyVString(batch[2], /*min_length=*/10, /*max_length=*/40);
    VerifyNationKey(batch[3]);
    VerifyPhone(batch[4]);
    VerifyDecimalsBetween(batch[5], -99999, 999999);
    CountModifiedComments(batch[6], &good_count, &bad_count);
    num_rows += batch.length;
  }
  ASSERT_EQ(seen_suppkey.size(), kExpectedRows);
  ASSERT_EQ(num_rows, kExpectedRows);
  ASSERT_EQ(good_count, static_cast<int64_t>(5 * scale_factor));
  ASSERT_EQ(bad_count, static_cast<int64_t>(5 * scale_factor));
}

TEST(TpchNode, Supplier) {
  ASSERT_OK_AND_ASSIGN(auto res, GenerateTable(&TpchGen::Supplier));