diff options
Diffstat (limited to 'test/std/numerics/numeric.ops/transform.reduce/transform_reduce_iter_iter_iter_init_op_op.pass.cpp')
| -rw-r--r-- | test/std/numerics/numeric.ops/transform.reduce/transform_reduce_iter_iter_iter_init_op_op.pass.cpp | 97 |
1 files changed, 97 insertions, 0 deletions
diff --git a/test/std/numerics/numeric.ops/transform.reduce/transform_reduce_iter_iter_iter_init_op_op.pass.cpp b/test/std/numerics/numeric.ops/transform.reduce/transform_reduce_iter_iter_iter_init_op_op.pass.cpp new file mode 100644 index 000000000000..b1b53293b991 --- /dev/null +++ b/test/std/numerics/numeric.ops/transform.reduce/transform_reduce_iter_iter_iter_init_op_op.pass.cpp @@ -0,0 +1,97 @@ +//===----------------------------------------------------------------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is dual licensed under the MIT and the University of Illinois Open +// Source Licenses. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +// <numeric> +// UNSUPPORTED: c++98, c++03, c++11, c++14 + +// template <class InputIterator1, class InputIterator2, class T, +// class BinaryOperation1, class BinaryOperation2> +// T transform_reduce(InputIterator1 first1, InputIterator1 last1, +// InputIterator2 first2, T init, +// BinaryOperation1 binary_op1, BinaryOperation2 binary_op2); +// + +#include <numeric> +#include <cassert> + +#include "test_iterators.h" + +template <class Iter1, class Iter2, class T, class Op1, class Op2> +void +test(Iter1 first1, Iter1 last1, Iter2 first2, T init, Op1 op1, Op2 op2, T x) +{ + static_assert( std::is_same_v<T, + decltype(std::transform_reduce(first1, last1, first2, init, op1, op2))> ); + assert(std::transform_reduce(first1, last1, first2, init, op1, op2) == x); +} + +template <class SIter, class UIter> +void +test() +{ + int ia[] = {1, 2, 3, 4, 5, 6}; + unsigned int ua[] = {2, 4, 6, 8, 10,12}; + unsigned sa = sizeof(ia) / sizeof(ia[0]); + assert(sa == sizeof(ua) / sizeof(ua[0])); // just to be sure + + test(SIter(ia), SIter(ia), UIter(ua), 0, std::plus<>(), std::multiplies<>(), 0); + test(UIter(ua), UIter(ua), SIter(ia), 1, std::multiplies<>(), std::plus<>(), 1); + test(SIter(ia), SIter(ia+1), UIter(ua), 0, std::multiplies<>(), std::plus<>(), 0); + test(UIter(ua), UIter(ua+1), SIter(ia), 2, std::plus<>(), std::multiplies<>(), 4); + test(SIter(ia), SIter(ia+2), UIter(ua), 0, std::plus<>(), std::multiplies<>(), 10); + test(UIter(ua), UIter(ua+2), SIter(ia), 3, std::multiplies<>(), std::plus<>(), 54); + test(SIter(ia), SIter(ia+sa), UIter(ua), 4, std::multiplies<>(), std::plus<>(), 2099520); + test(UIter(ua), UIter(ua+sa), SIter(ia), 4, std::plus<>(), std::multiplies<>(), 186); +} + +template <typename T, typename Init> +void test_return_type() +{ + T *p = nullptr; + static_assert( std::is_same_v<Init, + decltype(std::transform_reduce(p, p, p, Init{}, std::plus<>(), std::multiplies<>()))> ); +} + +int main() +{ + test_return_type<char, int>(); + test_return_type<int, int>(); + test_return_type<int, unsigned long>(); + test_return_type<float, int>(); + test_return_type<short, float>(); + test_return_type<double, char>(); + test_return_type<char, double>(); + +// All the iterator categories + test<input_iterator <const int*>, input_iterator <const unsigned int*> >(); + test<input_iterator <const int*>, forward_iterator <const unsigned int*> >(); + test<input_iterator <const int*>, bidirectional_iterator<const unsigned int*> >(); + test<input_iterator <const int*>, random_access_iterator<const unsigned int*> >(); + + test<forward_iterator <const int*>, input_iterator <const unsigned int*> >(); + test<forward_iterator <const int*>, forward_iterator <const unsigned int*> >(); + test<forward_iterator <const int*>, bidirectional_iterator<const unsigned int*> >(); + test<forward_iterator <const int*>, random_access_iterator<const unsigned int*> >(); + + test<bidirectional_iterator<const int*>, input_iterator <const unsigned int*> >(); + test<bidirectional_iterator<const int*>, forward_iterator <const unsigned int*> >(); + test<bidirectional_iterator<const int*>, bidirectional_iterator<const unsigned int*> >(); + test<bidirectional_iterator<const int*>, random_access_iterator<const unsigned int*> >(); + + test<random_access_iterator<const int*>, input_iterator <const unsigned int*> >(); + test<random_access_iterator<const int*>, forward_iterator <const unsigned int*> >(); + test<random_access_iterator<const int*>, bidirectional_iterator<const unsigned int*> >(); + test<random_access_iterator<const int*>, random_access_iterator<const unsigned int*> >(); + +// just plain pointers (const vs. non-const, too) + test<const int*, const unsigned int *>(); + test<const int*, unsigned int *>(); + test< int*, const unsigned int *>(); + test< int*, unsigned int *>(); +} |