mirror of
https://github.com/llvm-mirror/libcxx.git
synced 2025-10-23 10:07:41 +08:00

LLVM uses .h as its extension for header files. Files renamed using: for f in libcxx/test/support/*.hpp; do git mv $f ${f%.hpp}.h; done References to the files updated using: for f in $(git diff master | grep 'rename from' | cut -f 3 -d ' '); do a=$(basename $f); echo $a; rg -l $a libcxx | xargs sed -i '' "s/$a/${a%.hpp}.h/"; done HPP include guards updated manually using: for f in $(git diff master | grep 'rename from' | cut -f 3 -d ' '); do echo ${f%.hpp}.h ; done | xargs mvim Differential Revision: https://reviews.llvm.org/D66104 git-svn-id: https://llvm.org/svn/llvm-project/libcxx/trunk@369481 91177308-0d34-0410-b5e6-96231b3b80d8
173 lines
6.8 KiB
C++
173 lines
6.8 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>
|
|
|
|
// template<class ForwardIterator, class Size, class T, class BinaryPredicate>
|
|
// constexpr ForwardIterator // constexpr after C++17
|
|
// search_n(ForwardIterator first, ForwardIterator last, Size count,
|
|
// const T& value, BinaryPredicate pred);
|
|
|
|
#include <algorithm>
|
|
#include <cassert>
|
|
|
|
#include "test_macros.h"
|
|
#include "test_iterators.h"
|
|
#include "user_defined_integral.h"
|
|
|
|
#if TEST_STD_VER > 17
|
|
TEST_CONSTEXPR bool eq(int a, int b) { return a == b; }
|
|
|
|
TEST_CONSTEXPR bool test_constexpr() {
|
|
int ia[] = {0, 0, 1, 1, 2, 2};
|
|
return (std::search_n(std::begin(ia), std::end(ia), 1, 0, eq) == ia)
|
|
&& (std::search_n(std::begin(ia), std::end(ia), 2, 1, eq) == ia+2)
|
|
&& (std::search_n(std::begin(ia), std::end(ia), 1, 3, eq) == std::end(ia))
|
|
;
|
|
}
|
|
#endif
|
|
|
|
struct count_equal
|
|
{
|
|
static unsigned count;
|
|
template <class T>
|
|
bool operator()(const T& x, const T& y)
|
|
{++count; return x == y;}
|
|
};
|
|
|
|
unsigned count_equal::count = 0;
|
|
|
|
|
|
template <class Iter>
|
|
void
|
|
test()
|
|
{
|
|
int ia[] = {0, 1, 2, 3, 4, 5};
|
|
const unsigned sa = sizeof(ia)/sizeof(ia[0]);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 0, 0, count_equal()) == Iter(ia));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 1, 0, count_equal()) == Iter(ia+0));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 2, 0, count_equal()) == Iter(ia+sa));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), sa, 0, count_equal()) == Iter(ia+sa));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 0, 3, count_equal()) == Iter(ia));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 1, 3, count_equal()) == Iter(ia+3));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 2, 3, count_equal()) == Iter(ia+sa));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), sa, 3, count_equal()) == Iter(ia+sa));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 0, 5, count_equal()) == Iter(ia));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 1, 5, count_equal()) == Iter(ia+5));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), 2, 5, count_equal()) == Iter(ia+sa));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ia), Iter(ia+sa), sa, 5, count_equal()) == Iter(ia+sa));
|
|
assert(count_equal::count <= sa);
|
|
count_equal::count = 0;
|
|
|
|
int ib[] = {0, 0, 1, 1, 2, 2};
|
|
const unsigned sb = sizeof(ib)/sizeof(ib[0]);
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 0, 0, count_equal()) == Iter(ib));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 1, 0, count_equal()) == Iter(ib+0));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 2, 0, count_equal()) == Iter(ib+0));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 3, 0, count_equal()) == Iter(ib+sb));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), sb, 0, count_equal()) == Iter(ib+sb));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 0, 1, count_equal()) == Iter(ib));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 1, 1, count_equal()) == Iter(ib+2));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 2, 1, count_equal()) == Iter(ib+2));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 3, 1, count_equal()) == Iter(ib+sb));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), sb, 1, count_equal()) == Iter(ib+sb));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 0, 2, count_equal()) == Iter(ib));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 1, 2, count_equal()) == Iter(ib+4));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 2, 2, count_equal()) == Iter(ib+4));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), 3, 2, count_equal()) == Iter(ib+sb));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ib), Iter(ib+sb), sb, 2, count_equal()) == Iter(ib+sb));
|
|
assert(count_equal::count <= sb);
|
|
count_equal::count = 0;
|
|
|
|
int ic[] = {0, 0, 0};
|
|
const unsigned sc = sizeof(ic)/sizeof(ic[0]);
|
|
assert(std::search_n(Iter(ic), Iter(ic+sc), 0, 0, count_equal()) == Iter(ic));
|
|
assert(count_equal::count <= sc);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ic), Iter(ic+sc), 1, 0, count_equal()) == Iter(ic));
|
|
assert(count_equal::count <= sc);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ic), Iter(ic+sc), 2, 0, count_equal()) == Iter(ic));
|
|
assert(count_equal::count <= sc);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ic), Iter(ic+sc), 3, 0, count_equal()) == Iter(ic));
|
|
assert(count_equal::count <= sc);
|
|
count_equal::count = 0;
|
|
assert(std::search_n(Iter(ic), Iter(ic+sc), 4, 0, count_equal()) == Iter(ic+sc));
|
|
assert(count_equal::count <= sc);
|
|
count_equal::count = 0;
|
|
|
|
// Check that we properly convert the size argument to an integral.
|
|
TEST_IGNORE_NODISCARD std::search_n(Iter(ic), Iter(ic+sc), UserDefinedIntegral<unsigned>(4), 0, count_equal());
|
|
count_equal::count = 0;
|
|
}
|
|
|
|
int main(int, char**)
|
|
{
|
|
test<forward_iterator<const int*> >();
|
|
test<bidirectional_iterator<const int*> >();
|
|
test<random_access_iterator<const int*> >();
|
|
|
|
#if TEST_STD_VER > 17
|
|
static_assert(test_constexpr());
|
|
#endif
|
|
|
|
return 0;
|
|
}
|