Function VerifyOrders

cpp/src/arrow/acero/tpch_node_test.cc:470–502

}

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

  std::unordered_set<int32_t> seen_orderkey;
  for (auto& batch : batches) {
    ValidateBatch(batch);
    VerifyUniqueKey(&seen_orderkey, batch[0],
                    /*min=*/1,
                    /*max=*/static_cast<int32_t>(4 * kExpectedRows));
    VerifyAllBetween(batch[1], /*min=*/1, /*max=*/static_cast<int32_t>(kExpectedRows));
    VerifyModuloBetween(batch[1], /*min=*/1, /*max=*/2, /*mod=*/3);
    VerifyOneOf(batch[2], {'F', 'O', 'P'});
    VerifyAllBetween(batch[4], kStartDate, kEndDate - 151);
    VerifyOneOf(batch[5],
                /*byte_width=*/15,
                {
                    "1-URGENT",
                    "2-HIGH",
                    "3-MEDIUM",
                    "4-NOT SPECIFIED",
                    "5-LOW",
                });
    VerifyStringAndNumber_FixedWidth(batch[6], Datum(),
                                     /*byte_width=*/15, "Clerk#",
                                     /*verify_padding=*/true);
    VerifyAllBetween(batch[7], /*min=*/0, /*max=*/0);
    num_rows += batch.length;
  }
  ASSERT_EQ(seen_orderkey.size(), kExpectedRows);
  ASSERT_EQ(num_rows, kExpectedRows);
}

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