Function VerifyCustomer

cpp/src/arrow/acero/tpch_node_test.cc:440–463

}

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

  std::unordered_set<int32_t> seen_custkey;
  for (auto& batch : batches) {
    ValidateBatch(batch);
    VerifyUniqueKey(&seen_custkey, batch[0],
                    /*min=*/1,
                    /*max=*/static_cast<int32_t>(kExpectedRows));
    VerifyStringAndNumber_Varlen(batch[1], batch[0], "Customer#");
    VerifyVString(batch[2], /*min=*/10, /*max=*/40);
    VerifyNationKey(batch[3]);
    VerifyPhone(batch[4]);
    VerifyDecimalsBetween(batch[5], -99999, 999999);
    VerifyCorrectNumberOfWords_FixedWidth(batch[6],
                                          /*num_words=*/1,
                                          /*byte_width=*/10);
    num_rows += batch.length;
  }
  ASSERT_EQ(seen_custkey.size(), kExpectedRows);
  ASSERT_EQ(num_rows, kExpectedRows);
}

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