Function VerifyPart

cpp/src/arrow/acero/tpch_node_test.cc:378–407

}

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

  std::unordered_set<int32_t> seen_partkey;
  for (auto& batch : batches) {
    ValidateBatch(batch);
    VerifyUniqueKey(&seen_partkey, batch[0],
                    /*min=*/1,
                    /*max=*/static_cast<int32_t>(kExpectedRows));
    VerifyCorrectNumberOfWords_Varlen(batch[1],
                                      /*num_words*=*/5);
    VerifyStringAndNumber_FixedWidth(batch[2], Datum(),
                                     /*byte_width=*/25, "Manufacturer#",
                                     /*verify_padding=*/false);
    VerifyStringAndNumber_FixedWidth(batch[3], Datum(),
                                     /*byte_width=*/10, "Brand#",
                                     /*verify_padding=*/false);
    VerifyCorrectNumberOfWords_Varlen(batch[4],
                                      /*num_words=*/3);
    VerifyAllBetween(batch[5], /*min=*/1, /*max=*/50);
    VerifyCorrectNumberOfWords_FixedWidth(batch[6],
                                          /*num_words=*/2,
                                          /*byte_width=*/10);
    num_rows += batch.length;
  }
  ASSERT_EQ(seen_partkey.size(), kExpectedRows);
  ASSERT_EQ(num_rows, kExpectedRows);
}

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