libstdc++-v3/testsuite/25_algorithms/generate_n/87982.cc

   1 // Copyright (C) 2019-2025 Free Software Foundation, Inc.
   2 //
   3 // This file is part of the GNU ISO C++ Library.  This library is free
   4 // software; you can redistribute it and/or modify it under the
   5 // terms of the GNU General Public License as published by the
   6 // Free Software Foundation; either version 3, or (at your option)
   7 // any later version.
   8
   9 // This library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13
  14 // You should have received a copy of the GNU General Public License along
  15 // with this library; see the file COPYING3.  If not see
  16 // <http://www.gnu.org/licenses/>.
  17
  18 // { dg-do run }
  19
  20 #include <algorithm>
  21 #include <testsuite_hooks.h>
  22
  23 int a[4] = { 0, 1, 2, 3 };
  24 int g;
  25 int gen() { return g; }
  26 enum E { e2 = 2 };
  27
  28 struct Num
  29 {
  30   char val;
  31
  32   operator char() const { return val; }
  33
  34 private:
  35   void operator+() const;
  36   void operator+(int) const;
  37   void operator+(Num) const;
  38   void operator<(int) const;
  39   void operator>(int) const;
  40   void operator<=(int) const;
  41   void operator>=(int) const;
  42   void operator==(int) const;
  43   void operator!=(int) const;
  44 };
  45
  46 void
  47 test01()
  48 {
  49   int* p;
  50
  51   g = -1;
  52   p = std::generate_n(a, true, &gen); // bool as Size
  53   VERIFY( p == a+1 );
  54   VERIFY( a[0] == g );
  55   VERIFY( a[1] == 1 );
  56   VERIFY( a[2] == 2 );
  57   VERIFY( a[3] == 3 );
  58
  59   g = -2;
  60   p = std::generate_n(a, e2, &gen); // enumeration type as Size
  61   VERIFY( p == a+2 );
  62   VERIFY( a[0] == g );
  63   VERIFY( a[1] == g );
  64   VERIFY( a[2] == 2 );
  65   VERIFY( a[3] == 3 );
  66
  67   g = -3;
  68   p = std::generate_n(a, 3.5, &gen); // floating point type as Size
  69   VERIFY( p == a+3 );
  70   VERIFY( a[0] == g );
  71   VERIFY( a[1] == g );
  72   VERIFY( a[2] == g );
  73   VERIFY( a[3] == 3 );
  74
  75   g = -4;
  76   Num n = { 3 };
  77   p = std::generate_n(a, n, &gen); // non-scalar type as Size
  78   VERIFY( p == a+3 );
  79   VERIFY( a[0] == g );
  80   VERIFY( a[1] == g );
  81   VERIFY( a[2] == g );
  82   VERIFY( a[3] == 3 );
  83 }
  84
  85 int main()
  86 {
  87   test01();
  88 }