Method CheckReserve

cpp/src/arrow/util/small_vector_test.cc:198–225

  template <size_t N>
  void CheckReserve(size_t max_size, bool expect_overflow) {
    IntVectorType<N> ints;
    ints.emplace_back(123);

    size_t orig_capacity = ints.capacity();

    ints.reserve(max_size / 3);
    ASSERT_EQ(ints.capacity(), std::max(max_size / 3, orig_capacity));
    ASSERT_EQ(ints.size(), 1);
    ASSERT_EQ(ints[0], 123);

    ints.reserve(4 * max_size / 5);
    ASSERT_EQ(ints.capacity(), std::max(4 * max_size / 5, orig_capacity));
    ASSERT_EQ(ints.size(), 1);
    ASSERT_EQ(ints[0], 123);
    ASSERT_EQ(UsesStaticStorage(ints), !expect_overflow);

    size_t old_capacity = ints.capacity();
    ints.reserve(max_size / 5);  // no-op
    ASSERT_EQ(ints.capacity(), old_capacity);
    ASSERT_EQ(ints.size(), 1);
    ASSERT_EQ(ints[0], 123);

    ints.reserve(1);  // no-op
    ASSERT_EQ(ints.capacity(), old_capacity);
    ASSERT_EQ(ints.size(), 1);
    ASSERT_EQ(ints[0], 123);
  }

  void TestReserve() {
    CheckReserve<Traits::TestSizeFor(12)>(12, /*expect_overflow=*/Traits::CanOverflow());