libcxx/test/std/containers/sequences/vector/vector.capacity/reserve.pass.cpp

   1 //===----------------------------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 // <vector>
  10
  11 // void reserve(size_type n);
  12
  13 #include <vector>
  14 #include <cassert>
  15 #include "test_macros.h"
  16 #include "test_allocator.h"
  17 #include "min_allocator.h"
  18 #include "asan_testing.h"
  19
  20 int main(int, char**)
  21 {
  22     {
  23         std::vector<int> v;
  24         v.reserve(10);
  25         assert(v.capacity() >= 10);
  26         assert(is_contiguous_container_asan_correct(v));
  27     }
  28     {
  29         std::vector<int> v(100);
  30         assert(v.capacity() == 100);
  31         v.reserve(50);
  32         assert(v.size() == 100);
  33         assert(v.capacity() == 100);
  34         v.reserve(150);
  35         assert(v.size() == 100);
  36         assert(v.capacity() == 150);
  37         assert(is_contiguous_container_asan_correct(v));
  38     }
  39     {
  40         // Add 1 for implementations that dynamically allocate a container proxy.
  41         std::vector<int, limited_allocator<int, 250 + 1> > v(100);
  42         assert(v.capacity() == 100);
  43         v.reserve(50);
  44         assert(v.size() == 100);
  45         assert(v.capacity() == 100);
  46         v.reserve(150);
  47         assert(v.size() == 100);
  48         assert(v.capacity() == 150);
  49         assert(is_contiguous_container_asan_correct(v));
  50     }
  51 #if TEST_STD_VER >= 11
  52     {
  53         std::vector<int, min_allocator<int>> v;
  54         v.reserve(10);
  55         assert(v.capacity() >= 10);
  56         assert(is_contiguous_container_asan_correct(v));
  57     }
  58     {
  59         std::vector<int, min_allocator<int>> v(100);
  60         assert(v.capacity() == 100);
  61         v.reserve(50);
  62         assert(v.size() == 100);
  63         assert(v.capacity() == 100);
  64         v.reserve(150);
  65         assert(v.size() == 100);
  66         assert(v.capacity() == 150);
  67         assert(is_contiguous_container_asan_correct(v));
  68     }
  69 #endif
  70
  71   return 0;
  72 }