File: sequence.cpp

package info (click to toggle)
zug 0.1.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,052 kB
  • sloc: cpp: 6,168; makefile: 203; sh: 88; python: 62
file content (118 lines) | stat: -rw-r--r-- 3,160 bytes parent folder | download
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
//
// zug: transducers for C++
// Copyright (C) 2019 Juan Pedro Bolivar Puente
//
// This software is distributed under the Boost Software License, Version 1.0.
// See accompanying file LICENSE or copy at http://boost.org/LICENSE_1_0.txt
//

#include <catch2/catch.hpp>

#include <zug/compose.hpp>
#include <zug/sequence.hpp>
#include <zug/transducer/enumerate.hpp>
#include <zug/transducer/filter.hpp>
#include <zug/transducer/interleave.hpp>
#include <zug/transducer/map.hpp>
#include <zug/transducer/take.hpp>

#include <algorithm>
#include <vector>

using namespace zug;

namespace {
template <class RangeT, class OutputIt>
void my_copy(RangeT&& range, OutputIt d_first)
{
    std::copy(std::begin(range), std::end(range), d_first);
}
} // namespace

TEST_CASE("sequence, identity")
{
    auto v   = std::vector<int>{1, 2, 3, 4};
    auto r   = std::vector<int>{};
    auto seq = sequence(identity, v);
    my_copy(seq, std::back_inserter(r));
    CHECK(r == v);
}

TEST_CASE("sequence, filter_map")
{
    auto v   = std::vector<int>{1, 2, 3, 4};
    auto r   = std::vector<int>{};
    auto seq = sequence<int>(comp(filter([](int x) { return x % 2 == 0; }),
                                  map([](int x) { return x * 2; })),
                             v);
    my_copy(seq, std::back_inserter(r));
    CHECK(r == (std::vector<int>{{4, 8}}));
}

TEST_CASE("sequence, rvalues")
{
    auto r = std::vector<int>{};
    my_copy(sequence(identity, std::vector<int>{1, 2, 3, 4}),
            std::back_inserter(r));
    CHECK(r == (std::vector<int>{1, 2, 3, 4}));
}

TEST_CASE("sequence, constant")
{
    const auto v = std::vector<int>{1, 2, 3, 4};
    auto r       = std::vector<int>{};
    my_copy(sequence(identity, v), std::back_inserter(r));
    CHECK(r == v);
}

TEST_CASE("sequence, no shared iterator state")
{
    const auto v = std::vector<int>{1, 2, 3, 4};
    auto seq     = sequence(take(3), v);

    auto f2 = seq.begin();
    auto e  = seq.end();
    auto f1 = f2++;

    auto r1 = std::vector<int>{};
    std::copy(f1, e, std::back_inserter(r1));
    CHECK(r1 == (decltype(r1){1, 2, 3}));

    auto r2 = std::vector<int>{};
    std::copy(f2, e, std::back_inserter(r2));
    CHECK(r2 == (decltype(r2){2, 3}));
}

TEST_CASE("sequence, empty")
{
    const auto v = std::vector<int>{};
    auto seq     = sequence(take(3), v);
    auto r       = std::vector<int>{};
    my_copy(seq, std::back_inserter(r));
    CHECK(r == (decltype(r){}));
}

TEST_CASE("sequence, variadic growing no type deduction")
{
    using res_t = ZUG_VARIANT<int, std::string>;

    const auto v1 = std::vector<int>{1, 2, 3, 4};
    const auto v2 = std::vector<std::string>{"a", "b", "c"};
    auto seq      = sequence<res_t>(interleave, v1, v2);

    auto r = std::vector<res_t>{};
    my_copy(seq, std::back_inserter(r));
    CHECK(
        r ==
        (decltype(r){
            res_t{1}, res_t{"a"}, res_t{2}, res_t{"b"}, res_t{3}, res_t{"c"}}));
}

TEST_CASE("sequence, generator")
{
    auto seq = sequence(enumerate);

    auto r = std::vector<std::size_t>{};
    std::copy_n(seq.begin(), 4u, std::back_inserter(r));
    CHECK(r == (decltype(r){0u, 1u, 2u, 3u}));
}