Nikolas Klauser ade9c3bdca [libc++][PSTL] Implement std::fill{,_n}
Reviewed By: ldionne, #libc

Spies: libcxx-commits

Differential Revision: https://reviews.llvm.org/D149540
2023-05-01 14:03:40 -07:00

83 lines
2.5 KiB
C++

//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
// <algorithm>
// UNSUPPORTED: c++03, c++11, c++14
// REQUIRES: with-pstl
// template<class ExecutionPolicy, class ForwardIterator, class T>
// void fill(ExecutionPolicy&& exec,
// ForwardIterator first, ForwardIterator last, const T& value);
#include <algorithm>
#include <cassert>
#include <vector>
#include "test_macros.h"
#include "test_execution_policies.h"
#include "test_iterators.h"
EXECUTION_POLICY_SFINAE_TEST(fill);
static_assert(sfinae_test_fill<int, int*, int*, bool (*)(int)>);
static_assert(!sfinae_test_fill<std::execution::parallel_policy, int*, int*, int>);
template <class Iter>
struct Test {
template <class Policy>
void operator()(Policy&& policy) {
{ // simple test
int a[4];
std::fill(policy, Iter(std::begin(a)), Iter(std::end(a)), 33);
assert(std::all_of(std::begin(a), std::end(a), [](int i) { return i == 33; }));
}
{ // check that an empty range works
int a[1] = {2};
std::fill(policy, Iter(std::begin(a)), Iter(std::begin(a)), 33);
assert(a[0] == 2);
}
{ // check that a one-element range works
int a[1];
std::fill(policy, Iter(std::begin(a)), Iter(std::end(a)), 33);
assert(std::all_of(std::begin(a), std::end(a), [](int i) { return i == 33; }));
}
{ // check that a two-element range works
int a[2];
std::fill(policy, Iter(std::begin(a)), Iter(std::end(a)), 33);
assert(std::all_of(std::begin(a), std::end(a), [](int i) { return i == 33; }));
}
{ // check that a large range works
std::vector<int> a(234, 2);
std::fill(policy, Iter(std::data(a)), Iter(std::data(a) + std::size(a)), 33);
assert(std::all_of(std::begin(a), std::end(a), [](int i) { return i == 33; }));
}
}
};
struct ThrowOnCopy {
ThrowOnCopy& operator=(const ThrowOnCopy&) { throw int{}; }
};
int main(int, char**) {
types::for_each(types::forward_iterator_list<int*>{}, TestIteratorWithPolicies<Test>{});
#ifndef TEST_HAS_NO_EXCEPTIONS
std::set_terminate(terminate_successful);
ThrowOnCopy a[2];
try {
(void)std::fill(std::execution::par, std::begin(a), std::end(a), ThrowOnCopy{});
} catch (int) {
assert(false);
}
#endif
return 0;
}