Function MakeStruct

cpp/src/arrow/compute/kernels/scalar_if_else_test.cc:1246–1458

}

Datum MakeStruct(const std::vector<Datum>& conds) {
  if (conds.size() == 0) {
    // The tests below want a struct scalar when no condition values passed,
    // not a StructArray of length 0
    ScalarVector value;
    return std::make_shared<StructScalar>(value, struct_({}));
  } else {
    EXPECT_OK_AND_ASSIGN(Datum result, CallFunction("make_struct", conds));
    return result;
  }
}

void TestCaseWhenFixedSize(const std::shared_ptr<DataType>& type) {
  auto cond_true = ScalarFromJSON(boolean(), "true");
  auto cond_false = ScalarFromJSON(boolean(), "false");
  auto cond_null = ScalarFromJSON(boolean(), "null");
  auto cond1 = ArrayFromJSON(boolean(), "[true, true, null, null]");
  auto cond2 = ArrayFromJSON(boolean(), "[true, false, true, null]");
  auto scalar_null = ScalarFromJSON(type, "null");
  auto values_null = ArrayFromJSON(type, "[null, null, null, null]");

  if (type->id() == Type::DATE64) {
    auto scalar1 = ScalarFromJSON(type, "86400000");
    auto scalar2 = ScalarFromJSON(type, "172800000");
    auto values1 = ArrayFromJSON(type, "[259200000, null, 432000000, 518400000]");
    auto values2 = ArrayFromJSON(type, "[604800000, 691200000, null, 864000000]");

    CheckScalar("case_when", {MakeStruct({}), values1}, values1);
    CheckScalar("case_when", {MakeStruct({}), values_null}, values_null);

    CheckScalar("case_when", {MakeStruct({cond_true}), scalar1, values1},
                *MakeArrayFromScalar(*scalar1, 4));
    CheckScalar("case_when", {MakeStruct({cond_false}), scalar1, values1}, values1);

    CheckScalar("case_when", {MakeStruct({cond_true}), values1}, values1);
    CheckScalar("case_when", {MakeStruct({cond_false}), values1}, values_null);
    CheckScalar("case_when", {MakeStruct({cond_null}), values1}, values_null);
    CheckScalar("case_when", {MakeStruct({cond_true}), values1, values2}, values1);
    CheckScalar("case_when", {MakeStruct({cond_false}), values1, values2}, values2);
    CheckScalar("case_when", {MakeStruct({cond_null}), values1, values2}, values2);

    CheckScalar("case_when", {MakeStruct({cond_true, cond_true}), values1, values2},
                values1);
    CheckScalar("case_when", {MakeStruct({cond_false, cond_false}), values1, values2},
                values_null);
    CheckScalar("case_when", {MakeStruct({cond_true, cond_false}), values1, values2},
                values1);
    CheckScalar("case_when", {MakeStruct({cond_false, cond_true}), values1, values2},
                values2);
    CheckScalar("case_when", {MakeStruct({cond_null, cond_true}), values1, values2},
                values2);
    CheckScalar("case_when",
                {MakeStruct({cond_false, cond_false}), values1, values2, values2},
                values2);

    CheckScalar("case_when", {MakeStruct({cond1, cond2}), scalar1, scalar2},
                ArrayFromJSON(type, "[86400000, 86400000, 172800000, null]"));
    CheckScalar("case_when", {MakeStruct({cond1}), scalar_null}, values_null);