Function VerifyLineitem

cpp/src/arrow/acero/tpch_node_test.cc:509–550

}

void VerifyLineitem(const std::vector<ExecBatch>& batches,
                    double scale_factor = kDefaultScaleFactor) {
  std::unordered_map<int32_t, int32_t> counts;
  for (auto& batch : batches) {
    ValidateBatch(batch);
    CountInstances(&counts, batch[0]);
    VerifyAllBetween(batch[1], /*min=*/1,
                     /*max=*/static_cast<int32_t>(200000 * scale_factor));
    VerifyAllBetween(batch[3], /*min=*/1, /*max=*/7);
    VerifyDecimalsBetween(batch[4], /*min=*/100, /*max=*/5000);
    VerifyDecimalsBetween(batch[6], /*min=*/0, /*max=*/10);
    VerifyDecimalsBetween(batch[7], /*min=*/0, /*max=*/8);
    VerifyOneOf(batch[8], {'R', 'A', 'N'});
    VerifyOneOf(batch[9], {'O', 'F'});
    VerifyAllBetween(batch[10], kStartDate + 1, kEndDate - 151 + 121);
    VerifyAllBetween(batch[11], kStartDate + 30, kEndDate - 151 + 90);
    VerifyAllBetween(batch[12], kStartDate + 2, kEndDate - 151 + 121 + 30);
    VerifyOneOf(batch[13],
                /*byte_width=*/25,
                {
                    "DELIVER IN PERSON",
                    "COLLECT COD",
                    "NONE",
                    "TAKE BACK RETURN",
                });
    VerifyOneOf(batch[14],
                /*byte_width=*/10,
                {
                    "REG AIR",
                    "AIR",
                    "RAIL",
                    "SHIP",
                    "TRUCK",
                    "MAIL",
                    "FOB",
                });
  }
  for (auto& count : counts) {
    ASSERT_GE(count.second, 1);
    ASSERT_LE(count.second, 7);
  }
}

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