summaryrefslogtreecommitdiff
path: root/libstdc++-v3/testsuite/25_algorithms/for_each/for_each_n.cc
blob: 073f81a7df641e6b7bc1fe1ed7c965c3a9bf4fdd (plain)
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
// { dg-options "-std=gnu++17" }
// { dg-do run { target c++17 } }

// Copyright (C) 2019-2020 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library.  This library is free
// software; you can redistribute it and/or modify it under the
// terms of the GNU General Public License as published by the
// Free Software Foundation; either version 3, or (at your option)
// any later version.

// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License along
// with this library; see the file COPYING3.  If not see
// <http://www.gnu.org/licenses/>.

#include <algorithm>
#include <testsuite_hooks.h>
#include <testsuite_iterators.h>

void test01()
{
  using __gnu_test::test_container;
  using __gnu_test::input_iterator_wrapper;
  int array[5] = { 1, 2, 3, 4, 5 };
  test_container<int, input_iterator_wrapper> con(array);

  int sum = 0;
  struct Func
  {
    Func(int& i) : i(i) { }
    Func(Func&&) = default;
    Func& operator=(Func&&) = delete;
    void operator()(int n) const { i += n; }
    int& i;
  };

  struct Size
  {
    Size(short v) : val(v) { }
    operator short() const { return val; }
    short val;
  };
  auto res = std::for_each_n(con.begin(), Size(con.size()), Func(sum));

  VERIFY( res == con.end() );
  VERIFY( sum == 15 );
}

void
test02()
{
  using __gnu_test::test_container;
  using __gnu_test::random_access_iterator_wrapper;
  int array[5] = { 2, 4, 6, 8, 10 };
  test_container<int, random_access_iterator_wrapper> con(array);

  int prod = 1;
  struct Func
  {
    Func(int& i) : i(i) { }
    Func(Func&&) = default;
    Func& operator=(Func&&) = delete;
    void operator()(int n) const { i *= n; }
    int& i;
  };

  struct Size
  {
    Size(short v) : val(v) { }
    operator short() const { return val; }
    short val;
  };
  auto res = std::for_each_n(con.begin(), Size(con.size()), Func(prod));

  VERIFY( res == con.end() );
  VERIFY( prod == 3840 );
}

int main()
{
  test01();
  test02();
}