Method Check

cpp/src/arrow/compute/kernels/aggregate_test.cc:969–975

  Datum Expected(int64_t value) { return MakeScalar(static_cast<int64_t>(value)); }

  void Check(Datum input, int64_t expected_all, bool has_nulls = true) {
    int64_t expected_valid = has_nulls ? expected_all - 1 : expected_all;
    int64_t expected_null = has_nulls ? 1 : 0;
    CheckScalar("count_distinct", {input}, Expected(expected_valid), &only_valid);
    CheckScalar("count_distinct", {input}, Expected(expected_null), &only_null);
    CheckScalar("count_distinct", {input}, Expected(expected_all), &all);
  }

  void Check(const std::shared_ptr<DataType>& type, std::string_view json,
             int64_t expected_all, bool has_nulls = true) {