libstdc++-v3/testsuite/23_containers/map/erasure.cc

   1 // { dg-options "-std=gnu++2a" }
   2 // { dg-do run { target c++2a } }
   3
   4 // Copyright (C) 2018-2019 Free Software Foundation, Inc.
   5 //
   6 // This file is part of the GNU ISO C++ Library.  This library is free
   7 // software; you can redistribute it and/or modify it under the
   8 // terms of the GNU General Public License as published by the
   9 // Free Software Foundation; either version 3, or (at your option)
  10 // any later version.
  11
  12 // This library is distributed in the hope that it will be useful,
  13 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 // GNU General Public License for more details.
  16
  17 // You should have received a copy of the GNU General Public License along
  18 // with this library; see the file COPYING3.  If not see
  19 // <http://www.gnu.org/licenses/>.
  20
  21 #include <map>
  22 #include <testsuite_hooks.h>
  23
  24 auto is_odd_pair = [](const std::pair<const int, std::string>& p)
  25 {
  26   return p.first % 2 != 0;
  27 };
  28
  29 void
  30 test01()
  31 {
  32   std::map<int, std::string> m{ { 10, "A" }, { 11, "B" },
  33                                 { 12, "C" }, { 14, "D" },
  34                                 { 15, "E" }, { 17, "F" },
  35                                 { 18, "G" }, { 19, "H" } };
  36   auto num = std::erase_if(m, is_odd_pair);
  37   std::map<int, std::string> t{ { 10, "A" }, { 12, "C" },
  38                                 { 14, "D" }, { 18, "G" } };
  39   VERIFY( m == t );
  40   VERIFY( num == 4 );
  41 }
  42
  43 void
  44 test02()
  45 {
  46   std::multimap<int, std::string> mm{ { 20, "S" }, { 21, "T" },
  47                                       { 22, "U" }, { 22, "V" },
  48                                       { 23, "W" }, { 23, "X" },
  49                                       { 24, "Y" }, { 25, "Z" } };
  50   auto num = std::erase_if(mm, is_odd_pair);
  51   std::multimap<int, std::string> t{ { 20, "S" }, { 22, "U" },
  52                                      { 22, "V" }, { 24, "Y" } };
  53   VERIFY( mm == t );
  54   VERIFY( num == 4 );
  55 }
  56
  57 int
  58 main()
  59 {
  60   test01();
  61   test02();
  62
  63   return 0;
  64 }