1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
// UNSUPPORTED: c++03, c++11, c++14, c++17, c++20
// ADDITIONAL_COMPILE_FLAGS(has-fconstexpr-steps): -fconstexpr-steps=2000000
// template<container-compatible-range<bool> R>
// constexpr void assign_range(R&& rg); // C++23
#include <sstream>
#include <vector>
#include "../insert_range_sequence_containers.h"
#include "test_macros.h"
// Tested cases:
// - different kinds of assignments (assigning an {empty/one-element/mid-sized/long range} to an
// {empty/one-element/full} container);
// - an exception is thrown when allocating new elements.
constexpr bool test() {
static_assert(test_constraints_assign_range<std::vector, bool, char>());
for_all_iterators_and_allocators<bool, const int*>([]<class Iter, class Sent, class Alloc>() {
test_sequence_assign_range<std::vector<bool, Alloc>, Iter, Sent>([]([[maybe_unused]] auto&& c) {
LIBCPP_ASSERT(c.__invariants());
// `is_contiguous_container_asan_correct` doesn't work on `vector<bool>`.
});
});
{ // Vector may or may not need to reallocate because of the assignment -- make sure to test both cases.
{ // Ensure reallocation happens. Note that `vector<bool>` typically reserves a lot of capacity.
constexpr int N = 255;
bool in[N] = {};
std::vector<bool> v = {0, 0, 0, 1, 1, 0, 0, 0};
assert(v.capacity() < v.size() + std::ranges::size(in));
v.assign_range(in);
assert(std::ranges::equal(v, in));
}
{ // Ensure no reallocation happens.
bool in[] = {1, 1, 0, 1, 1};
std::vector<bool> v = {0, 0, 0, 1, 1, 0, 0, 0};
v.assign_range(in);
assert(std::ranges::equal(v, in));
}
{ // Ensure input-only sized ranges are accepted.
using input_iter = cpp20_input_iterator<const bool*>;
const bool in[]{true, true, false, true};
std::vector<bool> v;
v.assign_range(std::views::counted(input_iter{std::ranges::begin(in)}, std::ranges::ssize(in)));
assert(std::ranges::equal(v, std::vector<bool>{true, true, false, true}));
}
}
return true;
}
#ifndef TEST_HAS_NO_LOCALIZATION
void test_counted_istream_view() {
std::istringstream is{"1 1 0 1"};
auto vals = std::views::istream<bool>(is);
std::vector<bool> v;
v.assign_range(std::views::counted(vals.begin(), 3));
assert(v == (std::vector{true, true, false}));
}
#endif
int main(int, char**) {
test();
static_assert(test());
// Note: `test_assign_range_exception_safety_throwing_copy` doesn't apply because copying booleans cannot throw.
test_assign_range_exception_safety_throwing_allocator<std::vector, bool>();
#ifndef TEST_HAS_NO_LOCALIZATION
test_counted_istream_view();
#endif
return 0;
}
|
