libstdc++-v3/testsuite/25_algorithms/partition_copy/1.cc

   1 // { dg-do run { target c++11 } }
   2
   3 // 2008-06-26  Paolo Carlini  <paolo.carlini@oracle.com>
   4
   5 // Copyright (C) 2008-2019 Free Software Foundation, Inc.
   6 //
   7 // This file is part of the GNU ISO C++ Library.  This library is free
   8 // software; you can redistribute it and/or modify it under the
   9 // terms of the GNU General Public License as published by the
  10 // Free Software Foundation; either version 3, or (at your option)
  11 // any later version.
  12
  13 // This library is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17
  18 // You should have received a copy of the GNU General Public License along
  19 // with this library; see the file COPYING3.  If not see
  20 // <http://www.gnu.org/licenses/>.
  21
  22 #include <algorithm>
  23 #include <testsuite_hooks.h>
  24 #include <testsuite_iterators.h>
  25
  26 using __gnu_test::test_container;
  27 using __gnu_test::input_iterator_wrapper;
  28 using __gnu_test::output_iterator_wrapper;
  29
  30 typedef test_container<int, input_iterator_wrapper> Icontainer;
  31 typedef test_container<int, output_iterator_wrapper> Ocontainer;
  32 int array[] = {0, 5, 2, 1, 3, 4};
  33
  34 bool
  35 pred(int i)
  36 { return i > 2; }
  37
  38 void
  39 test1()
  40 {
  41   int true_out[1] = { -1 };
  42   int false_out[1] = { -1 };
  43   Icontainer in_con(array, array);
  44   Ocontainer true_out_con(true_out, true_out);
  45   Ocontainer false_out_con(false_out, false_out);
  46
  47   std::pair<output_iterator_wrapper<int>, output_iterator_wrapper<int> > res =
  48     std::partition_copy(in_con.begin(), in_con.end(),
  49                         true_out_con.begin(), false_out_con.begin(), pred);
  50
  51   VERIFY( res.first.ptr == true_out );
  52   VERIFY( res.second.ptr == false_out );
  53 }
  54
  55 void
  56 test2()
  57 {
  58   int true_out[1] = { -1 };
  59   int false_out[1] = { -1 };
  60   Icontainer in_con(array, array + 2);
  61   Ocontainer true_out_con(true_out, true_out + 1);
  62   Ocontainer false_out_con(false_out, false_out + 1);
  63
  64   std::pair<output_iterator_wrapper<int>, output_iterator_wrapper<int> > res =
  65     std::partition_copy(in_con.begin(), in_con.end(),
  66                         true_out_con.begin(), false_out_con.begin(), pred);
  67
  68   VERIFY( res.first.ptr == true_out + 1 );
  69   VERIFY( res.second.ptr == false_out + 1 );
  70   VERIFY( true_out[0] == 5 );
  71   VERIFY( false_out[0] == 0 );
  72 }
  73
  74 void
  75 test3()
  76 {
  77   int true_out[3] = { -1, -1, -1 };
  78   int false_out[3] = { -1, -1, -1 };
  79   Icontainer in_con(array, array + 6);
  80   Ocontainer true_out_con(true_out, true_out + 3);
  81   Ocontainer false_out_con(false_out, false_out + 3);
  82
  83   std::pair<output_iterator_wrapper<int>, output_iterator_wrapper<int> > res =
  84     std::partition_copy(in_con.begin(), in_con.end(),
  85                         true_out_con.begin(), false_out_con.begin(), pred);
  86
  87   VERIFY( res.first.ptr == true_out + 3 );
  88   VERIFY( res.second.ptr == false_out + 3 );
  89   VERIFY( true_out[0] == 5 && true_out[1] == 3 && true_out[2] == 4 );
  90   VERIFY( false_out[0] == 0 && false_out[1] == 2 && false_out[2] == 1 );
  91 }
  92
  93 int
  94 main()
  95 {
  96   test1();
  97   test2();
  98   test3();
  99   return 0;
 100 }