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)}));
}
|