libcxx/test/std/containers/sequences/vector.bool/assign_range.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 // UNSUPPORTED: c++03, c++11, c++14, c++17, c++20
  10 // ADDITIONAL_COMPILE_FLAGS(has-fconstexpr-steps): -fconstexpr-steps=2000000
  11
  12 // template<container-compatible-range<bool> R>
  13 //   constexpr void assign_range(R&& rg); // C++23
  14
  15 #include <vector>
  16
  17 #include "../insert_range_sequence_containers.h"
  18 #include "test_macros.h"
  19
  20 // Tested cases:
  21 // - different kinds of assignments (assigning an {empty/one-element/mid-sized/long range} to an
  22 //   {empty/one-element/full} container);
  23 // - an exception is thrown when allocating new elements.
  24 constexpr bool test() {
  25   static_assert(test_constraints_assign_range<std::vector, bool, char>());
  26
  27   for_all_iterators_and_allocators<bool, const int*>([]<class Iter, class Sent, class Alloc>() {
  28     test_sequence_assign_range<std::vector<bool, Alloc>, Iter, Sent>([]([[maybe_unused]] auto&& c) {
  29       LIBCPP_ASSERT(c.__invariants());
  30       // `is_contiguous_container_asan_correct` doesn't work on `vector<bool>`.
  31     });
  32   });
  33
  34   { // Vector may or may not need to reallocate because of the assignment -- make sure to test both cases.
  35     { // Ensure reallocation happens. Note that `vector<bool>` typically reserves a lot of capacity.
  36       constexpr int N = 255;
  37       bool in[N] = {};
  38       std::vector<bool> v = {0, 0, 0, 1, 1, 0, 0, 0};
  39       assert(v.capacity() < v.size() + std::ranges::size(in));
  40
  41       v.assign_range(in);
  42       assert(std::ranges::equal(v, in));
  43     }
  44
  45     { // Ensure no reallocation happens.
  46       bool in[] = {1, 1, 0, 1, 1};
  47       std::vector<bool> v = {0, 0, 0, 1, 1, 0, 0, 0};
  48
  49       v.assign_range(in);
  50       assert(std::ranges::equal(v, in));
  51     }
  52   }
  53
  54   return true;
  55 }
  56
  57 int main(int, char**) {
  58   test();
  59   static_assert(test());
  60
  61   // Note: `test_assign_range_exception_safety_throwing_copy` doesn't apply because copying booleans cannot throw.
  62   test_assign_range_exception_safety_throwing_allocator<std::vector, bool>();
  63
  64   return 0;
  65 }