test/std/algorithms/alg.sorting/alg.binary.search/equal.range/equal_range_comp.pass.cpp

   1 //===----------------------------------------------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is dual licensed under the MIT and the University of Illinois Open
   6 // Source Licenses. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 // <algorithm>
  11
  12 // template<ForwardIterator Iter, class T, CopyConstructible Compare>
  13 //   requires Predicate<Compare, T, Iter::value_type>
  14 //         && Predicate<Compare, Iter::value_type, T>
  15 //   pair<Iter, Iter>
  16 //   equal_range(Iter first, Iter last, const T& value, Compare comp);
  17
  18 #include <algorithm>
  19 #include <functional>
  20 #include <vector>
  21 #include <cassert>
  22 #include <cstddef>
  23
  24 #include "test_iterators.h"
  25
  26 template <class Iter, class T>
  27 void
  28 test(Iter first, Iter last, const T& value)
  29 {
  30     std::pair<Iter, Iter> i = std::equal_range(first, last, value, std::greater<int>());
  31     for (Iter j = first; j != i.first; ++j)
  32         assert(std::greater<int>()(*j, value));
  33     for (Iter j = i.first; j != last; ++j)
  34         assert(!std::greater<int>()(*j, value));
  35     for (Iter j = first; j != i.second; ++j)
  36         assert(!std::greater<int>()(value, *j));
  37     for (Iter j = i.second; j != last; ++j)
  38         assert(std::greater<int>()(value, *j));
  39 }
  40
  41 template <class Iter>
  42 void
  43 test()
  44 {
  45     const unsigned N = 1000;
  46     const int M = 10;
  47     std::vector<int> v(N);
  48     int x = 0;
  49     for (std::size_t i = 0; i < v.size(); ++i)
  50     {
  51         v[i] = x;
  52         if (++x == M)
  53             x = 0;
  54     }
  55     std::sort(v.begin(), v.end(), std::greater<int>());
  56     for (x = 0; x <= M; ++x)
  57         test(Iter(v.data()), Iter(v.data()+v.size()), x);
  58 }
  59
  60 int main()
  61 {
  62     int d[] = {3, 2, 1, 0};
  63     for (int* e = d; e <= d+4; ++e)
  64         for (int x = -1; x <= 4; ++x)
  65             test(d, e, x);
  66
  67     test<forward_iterator<const int*> >();
  68     test<bidirectional_iterator<const int*> >();
  69     test<random_access_iterator<const int*> >();
  70     test<const int*>();
  71 }