Function TEST_F

cpp/src/arrow/record_batch_test.cc:65–105

class TestRecordBatch : public ::testing::Test {};

TEST_F(TestRecordBatch, Equals) {
  const int length = 10;

  auto f0 = field("f0", int32());
  auto f1 = field("f1", uint8());
  auto f2 = field("f2", int16());

  auto schema = ::arrow::schema({f0, f1, f2});
  auto schema_same = ::arrow::schema({f0, f1, f2});
  auto schema_fewer_fields = ::arrow::schema({f0, f1});

  random::RandomArrayGenerator gen(42);

  auto a_f0 = gen.ArrayOf(int32(), length);
  auto a_f1 = gen.ArrayOf(uint8(), length);
  auto a_f2 = gen.ArrayOf(int16(), length);
  auto a_f0_half = a_f0->Slice(0, length / 2);
  auto a_f1_half = a_f1->Slice(0, length / 2);
  auto a_f0_different = gen.ArrayOf(int32(), length);
  auto a_f1_different = gen.ArrayOf(uint8(), length);

  auto b = RecordBatch::Make(schema, length, {a_f0, a_f1, a_f2});
  auto b_same = RecordBatch::Make(schema_same, length, {a_f0, a_f1, a_f2});
  auto b_fewer_fields = RecordBatch::Make(schema_fewer_fields, length, {a_f0, a_f1});
  auto b_fewer_fields_half =
      RecordBatch::Make(schema_fewer_fields, length / 2, {a_f0_half, a_f1_half});
  auto b_fewer_fields_different =
      RecordBatch::Make(schema_fewer_fields, length, {a_f0_different, a_f1_different});

  // Same Values
  ASSERT_TRUE(b->Equals(*b_same));

  // Different number of columns
  ASSERT_FALSE(b->Equals(*b_fewer_fields));

  // Different number of rows
  ASSERT_FALSE(b_fewer_fields->Equals(*b_fewer_fields_half));

  // Different values
  ASSERT_FALSE(b_fewer_fields->Equals(*b_fewer_fields_different));
}

class TestRecordBatchEqualOptions : public TestRecordBatch {};