Function TEST_F

bolt/exec/tests/HashPartitionFunctionTest.cpp:44–134

};

TEST_F(HashPartitionFunctionTest, function) {
  const int numRows = 10'000;
  auto vector = makeRowVector(
      {makeFlatVector<int32_t>(numRows, [](auto row) { return row * 100 / 3; }),
       makeFlatVector<int32_t>(numRows, [](auto row) { return row * 128; })});
  auto rowType = asRowType(vector->type());

  // The test case the two hash partition functions having the same config.
  {
    std::vector<uint32_t> partitionsWithoutBits(numRows);
    HashPartitionFunction functionWithoutBits(4, rowType, {0});
    functionWithoutBits.partition(*vector, partitionsWithoutBits);
    EXPECT_EQ(4, functionWithoutBits.numPartitions());

    std::vector<uint32_t> partitionsWithBits(numRows);
    HashPartitionFunction functionWithBits(HashBitRange{0, 2}, rowType, {0});
    functionWithBits.partition(*vector, partitionsWithBits);
    EXPECT_EQ(partitionsWithoutBits, partitionsWithBits);
    EXPECT_EQ(4, functionWithBits.numPartitions());
  }

  // The test case the two hash partition functions with different configs.
  {
    std::vector<uint32_t> partitionsWithoutBits(numRows);
    HashPartitionFunction functionWithoutBits(4, rowType, {0});
    functionWithoutBits.partition(*vector, partitionsWithoutBits);
    EXPECT_EQ(4, functionWithoutBits.numPartitions());

    std::vector<uint32_t> partitionsWithBits(numRows);
    HashPartitionFunction functionWithBits(HashBitRange{0, 3}, rowType, {0});
    functionWithBits.partition(*vector, partitionsWithBits);
    EXPECT_NE(partitionsWithoutBits, partitionsWithBits);
    EXPECT_EQ(8, functionWithBits.numPartitions());
  }

  // The test case the two hash partition functions with different configs.
  {
    std::vector<uint32_t> partitionsWithoutBits(numRows);
    HashPartitionFunction functionWithoutBits(4, rowType, {0});
    functionWithoutBits.partition(*vector, partitionsWithoutBits);
    EXPECT_EQ(4, functionWithoutBits.numPartitions());

    std::vector<uint32_t> partitionsWithBits(numRows);
    HashPartitionFunction functionWithBits(HashBitRange{29, 31}, rowType, {0});
    functionWithBits.partition(*vector, partitionsWithBits);
    EXPECT_NE(partitionsWithoutBits, partitionsWithBits);
    EXPECT_EQ(4, functionWithBits.numPartitions());
  }

  // The test case the two hash partition functions with different configs.
  {
    std::vector<uint32_t> partitionsWithBits1(numRows);
    HashPartitionFunction functionWithBits1(HashBitRange{40, 42}, rowType, {0});
    functionWithBits1.partition(*vector, partitionsWithBits1);
    EXPECT_EQ(4, functionWithBits1.numPartitions());

    std::vector<uint32_t> partitionsWithBits2(numRows);
    HashPartitionFunction functionWithBits2(HashBitRange{29, 31}, rowType, {0});