File: fold.cpp

package info (click to toggle)
range-v3 0.12.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 5,652 kB
  • sloc: cpp: 76,839; xml: 226; sh: 89; python: 34; makefile: 19; perl: 15
file content (103 lines) | stat: -rw-r--r-- 3,371 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// Range v3 library
//
//  Copyright Eric Niebler 2014-present
//
//  Use, modification and distribution is subject to the
//  Boost Software License, Version 1.0. (See accompanying
//  file LICENSE_1_0.txt or copy at
//  http://www.boost.org/LICENSE_1_0.txt)
//
// Project home: https://github.com/ericniebler/range-v3
//

//===----------------------------------------------------------------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is dual licensed under the MIT and the University of Illinois Open
// Source Licenses. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#include <cmath>
#include <functional>

#include <range/v3/algorithm/fold.hpp>
#include <range/v3/algorithm/min.hpp>
#include <range/v3/core.hpp>

#include "../simple_test.hpp"
#include "../test_iterators.hpp"

struct Approx
{
    double value;
    Approx(double v)
      : value(v)
    {}

    friend bool operator==(Approx a, double d)
    {
        return std::fabs(a.value - d) < 0.001;
    }
    friend bool operator==(double d, Approx a)
    {
        return a == d;
    }
};

template<class Iter, class Sent = Iter>
void test_left()
{
    double da[] = {0.25, 0.75};
    CHECK(ranges::fold_left(Iter(da), Sent(da), 1, std::plus<>()) == Approx{1.0});
    CHECK(ranges::fold_left(Iter(da), Sent(da + 2), 1, std::plus<>()) == Approx{2.0});

    CHECK(ranges::fold_left_first(Iter(da), Sent(da), ranges::min) == ranges::nullopt);
    CHECK(ranges::fold_left_first(Iter(da), Sent(da + 2), ranges::min) ==
          ranges::optional<Approx>(0.25));

    using ranges::make_subrange;
    CHECK(ranges::fold_left(make_subrange(Iter(da), Sent(da)), 1, std::plus<>()) ==
          Approx{1.0});
    CHECK(ranges::fold_left(make_subrange(Iter(da), Sent(da + 2)), 1, std::plus<>()) ==
          Approx{2.0});
    CHECK(ranges::fold_left_first(make_subrange(Iter(da), Sent(da)), ranges::min) ==
          ranges::nullopt);
    CHECK(ranges::fold_left_first(make_subrange(Iter(da), Sent(da + 2)), ranges::min) ==
          ranges::optional<Approx>(0.25));
}

void test_right()
{
    double da[] = {0.25, 0.75};
    CHECK(ranges::fold_right(da, da + 2, 1, std::plus<>()) == Approx{2.0});
    CHECK(ranges::fold_right(da, 1, std::plus<>()) == Approx{2.0});

    // f(0.25, f(0.75, 1))
    CHECK(ranges::fold_right(da, da + 2, 1, std::minus<>()) == Approx{0.5});
    CHECK(ranges::fold_right(da, 1, std::minus<>()) == Approx{0.5});

    int xs[] = {1, 2, 3};
    auto concat = [](int i, std::string s) { return s + std::to_string(i); };
    CHECK(ranges::fold_right(xs, xs + 2, std::string(), concat) == "21");
    CHECK(ranges::fold_right(xs, std::string(), concat) == "321");
}

int main()
{
    test_left<InputIterator<const double *>>();
    test_left<ForwardIterator<const double *>>();
    test_left<BidirectionalIterator<const double *>>();
    test_left<RandomAccessIterator<const double *>>();
    test_left<const double *>();

    test_left<InputIterator<const double *>, Sentinel<const double *>>();
    test_left<ForwardIterator<const double *>, Sentinel<const double *>>();
    test_left<BidirectionalIterator<const double *>, Sentinel<const double *>>();
    test_left<RandomAccessIterator<const double *>, Sentinel<const double *>>();

    test_right();

    return ::test_result();
}