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

   1 // Copyright (C) 2005-2013 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 // 25.3.5.2 [lib.set.union]
  19
  20 #include <algorithm>
  21 #include <testsuite_hooks.h>
  22 #include <testsuite_iterators.h>
  23
  24 using __gnu_test::test_container;
  25 using __gnu_test::input_iterator_wrapper;
  26 using __gnu_test::output_iterator_wrapper;
  27 using std::set_union;
  28
  29 typedef test_container<int, input_iterator_wrapper> Icontainer;
  30 typedef test_container<int, output_iterator_wrapper> Ocontainer;
  31
  32 void
  33 test1()
  34 {
  35   int array1[1], array2[1];
  36   Icontainer con1(array1, array1);
  37   Icontainer con2(array1, array1);
  38   Ocontainer con3(array2, array2);
  39   VERIFY(set_union(con1.begin(), con1.end(), con2.begin(), con2.end(),
  40                    con3.begin()).ptr == array2);
  41 }
  42
  43 void
  44 test2()
  45 {
  46   int array1[] = {1};
  47   int array2[] = {0};
  48   Icontainer con1(array1, array1);
  49   Icontainer con2(array1, array1 + 1);
  50   Ocontainer con3(array2, array2 + 1);
  51   VERIFY(set_union(con1.begin(), con1.end(), con2.begin(), con2.end(),
  52                    con3.begin()).ptr == array2 + 1);
  53   VERIFY(array2[0] == 1);
  54 }
  55
  56 void
  57 test3()
  58 {
  59   int array1[] = {1};
  60   int array2[] = {0};
  61   Icontainer con1(array1, array1 + 1);
  62   Icontainer con2(array1, array1);
  63   Ocontainer con3(array2, array2 + 1);
  64   VERIFY(set_union(con1.begin(), con1.end(), con2.begin(), con2.end(),
  65                    con3.begin()).ptr == array2 + 1);
  66   VERIFY(array2[0] == 1);
  67 }
  68
  69 void
  70 test4()
  71 {
  72   int array1[]={0,1,1,2,4};
  73   int array2[]={1,2,3};
  74   int array3[6];
  75   Icontainer con1(array1, array1 + 5);
  76   Icontainer con2(array2, array2 + 3);
  77   Ocontainer con3(array3, array3 + 6);
  78   VERIFY(set_union(con1.begin(), con1.end(), con2.begin(), con2.end(),
  79                con3.begin()).ptr == array3 + 6);
  80   VERIFY(array3[0] == 0 && array3[1] == 1 && array3[2] == 1 &&
  81          array3[3] == 2 && array3[4] == 3 && array3[5] == 4);
  82 }
  83
  84 struct S
  85 {
  86   int i;
  87   int j;
  88   S() {}
  89   S(int in)
  90   {
  91     if(in > 0)
  92     {
  93       i = in;
  94       j = 1;
  95     }
  96     else
  97     {
  98       i = -in;
  99       j = 0;
 100     }
 101   }
 102 };
 103
 104 bool
 105 operator<(const S& s1, const S& s2)
 106 { return s1.i < s2.i; }
 107
 108 typedef test_container<S, input_iterator_wrapper> SIcontainer;
 109 typedef test_container<S, output_iterator_wrapper> SOcontainer;
 110
 111 void
 112 test5()
 113 {
 114   S array1[] = { -1, -1, -1, -2, -2, -4};
 115   S array2[] = { 1, 1, 1, 1, 2, 3, 4, 4};
 116   S array3[9];
 117   SIcontainer con1(array1, array1 + 6);
 118   SIcontainer con2(array2, array2 + 8);
 119   SOcontainer con3(array3, array3 + 9);
 120   VERIFY(set_union(con1.begin(), con1.end(), con2.begin(), con2.end(),
 121                    con3.begin()).ptr == array3 + 9);
 122   VERIFY(array3[0].j == 0 && array3[1].j == 0 && array3[2].j == 0 &&
 123          array3[3].j == 1 && array3[4].j == 0 && array3[5].j == 0 &&
 124          array3[6].j == 1 && array3[7].j == 0 && array3[8].j == 1);
 125 }
 126
 127 int
 128 main()
 129 {
 130   test1();
 131   test2();
 132   test3();
 133   test4();
 134   test5();
 135 }
 136