Function TEST_F

bolt/exec/tests/ConcatFilesSpillMergeStreamTest.cpp:249–282

} // namespace bytedance::bolt::exec::test

TEST_F(ConcatFilesSpillMergeStreamTest, stream) {
  struct {
    size_t maxOutputRows;
    size_t numStreams;

    std::string debugString() const {
      return fmt::format(
          "maxOutputRows:{} numStreams:{}", maxOutputRows, numStreams);
    }
  } testSettings[] = {
      {1, 1},
      {1, 3},
      {1, 8},
      {7, 1},
      {7, 3},
      {7, 8},
      {16, 1},
      {16, 3},
      {16, 8},
  };

  for (const auto& testData : testSettings) {
    SCOPED_TRACE(testData.debugString());
    auto [totalVectors, spillStreams] =
        generateInputs(testData.numStreams, testData.maxOutputRows);
    std::vector<RowVectorPtr> results = mergeSpillStreams(
        std::move(spillStreams), totalVectors.size(), testData.maxOutputRows);
    ASSERT_EQ(totalVectors.size(), results.size());
    const auto expectedResults =
        makeExpectedResults(totalVectors, testData.maxOutputRows);
    ASSERT_EQ(expectedResults.size(), results.size());
    ASSERT_TRUE(assertEqualResults(expectedResults, results));
  }
}