File: product.cpp

package info (click to toggle)
zug 0.1.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,076 kB
  • sloc: cpp: 6,209; makefile: 203; sh: 88; python: 62
file content (71 lines) | stat: -rw-r--r-- 1,982 bytes parent folder | download | duplicates (2)
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
//
// 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/into_vector.hpp>
#include <zug/sequence.hpp>
#include <zug/transducer/product.hpp>
#include <zug/transducer/range.hpp>
#include <zug/transducer/take.hpp>

using namespace zug;

TEST_CASE("product, product")
{
    // example1 {
    auto v1  = std::vector<int>{1, 2};
    auto v2  = std::vector<int>{3, 4};
    auto res = into_vector(product(v2), v1);
    using t  = std::vector<std::tuple<int, int>>;
    CHECK(res == t{{1, 3}, {1, 4}, {2, 3}, {2, 4}});
    // }
}

TEST_CASE("product, variadic")
{
    // example2 {
    auto v1  = std::vector<int>{1, 2};
    auto v2  = std::vector<int>{3, 4};
    auto v3  = std::vector<char>{'a', 'b'};
    auto res = into_vector(product(v2, v3), v1);
    using t  = std::vector<std::tuple<int, int, char>>;
    CHECK(res == t{{1, 3, 'a'},
                   {1, 3, 'b'},
                   {1, 4, 'a'},
                   {1, 4, 'b'},
                   {2, 3, 'a'},
                   {2, 3, 'b'},
                   {2, 4, 'a'},
                   {2, 4, 'b'}});
    // }
}

TEST_CASE("product, generator")
{
    using tup = std::tuple<int, int>;
    auto v1   = std::vector<int>{1, 2};
    auto v2   = std::vector<int>{4, 5};

    auto res = into_vector(comp(take(1), product(v1, v2)));
    CHECK(res == (decltype(res){tup(1, 4), tup(1, 5), tup(2, 4), tup(2, 5)}));
}

TEST_CASE("product, tranducer product example")
{
    using tup = std::tuple<int, int>;

    auto idx = sequence(range(2));
    auto res = into_vector(product(idx), idx);

    for (auto&& x : res) {
        std::cout << std::get<0>(x) << ", " << std::get<1>(x) << std::endl;
    }
    CHECK(res == (decltype(res){tup(0, 0), tup(0, 1), tup(1, 0), tup(1, 1)}));
}