Function TestCaseWhenRandom

cpp/src/arrow/compute/kernels/scalar_if_else_test.cc:1461–1511

}

void TestCaseWhenRandom(const std::shared_ptr<DataType>& type, int64_t len = 300) {
  random::RandomArrayGenerator rand(/*seed=*/0);

  // Adding 64 consecutive 1's and 0's in the cond array to test all-true/ all-false
  // word code paths
  ASSERT_OK_AND_ASSIGN(auto always_true, MakeArrayFromScalar(BooleanScalar(true), 64));
  ASSERT_OK_AND_ASSIGN(auto always_false, MakeArrayFromScalar(BooleanScalar(false), 64));
  auto maybe_true_with_nulls =
      rand.ArrayOf(boolean(), len - 64 * 2, /*null_probability=*/0.04);
  auto maybe_true_all_valid =
      rand.ArrayOf(boolean(), len - 64 * 2, /*null_probability=*/0.0);
  ASSERT_OK_AND_ASSIGN(auto concat1,
                       Concatenate({always_true, always_false, maybe_true_with_nulls}));
  auto cond1 = checked_pointer_cast<BooleanArray>(concat1);
  ASSERT_OK_AND_ASSIGN(auto concat2,
                       Concatenate({always_true, maybe_true_all_valid, always_false}));
  auto cond2 = checked_pointer_cast<BooleanArray>(concat2);

  auto value1 = rand.ArrayOf(type, len, /*null_probability=*/0.04);
  auto value2 = rand.ArrayOf(type, len, /*null_probability=*/0.04);
  auto value_else = rand.ArrayOf(type, len, /*null_probability=*/0.04);

  auto value1_span = ArraySpan(*value1->data());
  auto value2_span = ArraySpan(*value2->data());
  auto value_else_span = ArraySpan(*value_else->data());

  for (const bool has_else : {true, false}) {
    ASSERT_OK_AND_ASSIGN(auto builder, MakeBuilder(type));
    ASSERT_OK(builder->Reserve(len));
    for (int64_t i = 0; i < len; ++i) {
      if (cond1->IsValid(i) && cond1->Value(i)) {
        ASSERT_OK(builder->AppendArraySlice(value1_span, i, /*length=*/1));
      } else if (cond2->IsValid(i) && cond2->Value(i)) {
        ASSERT_OK(builder->AppendArraySlice(value2_span, i, /*length=*/1));
      } else if (has_else) {
        ASSERT_OK(builder->AppendArraySlice(value_else_span, i, /*length=*/1));
      } else {
        ASSERT_OK(builder->AppendNull());
      }
    }
    ASSERT_OK_AND_ASSIGN(auto expected_data, builder->Finish());

    if (has_else) {
      CheckScalar("case_when", {MakeStruct({cond1, cond2}), value1, value2, value_else},
                  expected_data);
    } else {
      CheckScalar("case_when", {MakeStruct({cond1, cond2}), value1, value2},
                  expected_data);
    }
  }
}

template <typename Type>
class TestCaseWhenNumeric : public ::testing::Test {};