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
|
// This source code is released into the public domain.
#include <catch2/catch_test_macros.hpp>
import nihil.std;
import nihil.core;
namespace {
inline auto constexpr test_tags = "[nihil][nihil.generator]";
SCENARIO("A generator that yields values", test_tags)
{
GIVEN ("A generator that yields values") {
auto fn = [&]() -> nihil::generator<int> {
co_yield 1;
co_yield 2;
};
THEN ("The generator yields the original values") {
REQUIRE(std::ranges::equal(fn(), std::vector{1, 2}));
}
}
}
SCENARIO("A generator that yields references", test_tags)
{
GIVEN ("A generator that yields references") {
auto one = 1, two = 2;
auto fn = [&]() -> nihil::generator<int &> {
co_yield one;
co_yield two;
};
auto range = fn();
THEN ("The references refer to the original values") {
auto it = std::ranges::begin(range);
REQUIRE(&*it == &one);
++it;
REQUIRE(&*it == &two);
++it;
REQUIRE(it == std::ranges::end(range));
}
}
}
SCENARIO("A generator that yields pointers", test_tags)
{
GIVEN ("A generator that yields pointers") {
auto one = 1, two = 2;
auto fn = [&]() -> nihil::generator<int *> {
co_yield &one;
co_yield &two;
};
THEN ("The pointers point to the original values") {
REQUIRE(std::ranges::equal(fn(), std::vector{&one, &two}));
}
}
}
SCENARIO("A generator that yields lvalues", test_tags)
{
GIVEN ("A generator that yields pointers") {
auto one = 1, two = 2;
auto fn = [&]() -> nihil::generator<int> {
co_yield one;
co_yield two;
};
THEN ("The pointers point to the original values") {
REQUIRE(std::ranges::equal(fn(), std::vector{1, 2}));
}
}
}
TEST_CASE("generator: exceptions", "[generator]")
{
auto fn = []() -> nihil::generator<int> {
co_yield 1;
throw std::runtime_error("test");
};
auto range = fn();
auto it = std::ranges::begin(range);
REQUIRE(*it == 1);
REQUIRE_THROWS_AS(it++, std::runtime_error);
}
#if 0
// TODO: Re-enable this test once we have a standard-compliant generator.
TEST_CASE("generator: elements_of", "[generator]")
{
auto fn1 = [] -> nihil::generator<int> {
co_yield 1;
co_yield 2;
co_yield 3;
};
auto fn2 = [&fn1] -> nihil::generator<int> {
co_yield nihil::elements_of(fn1());
};
auto values = std::vector<int>();
std::ranges::copy(fn2(), std::back_inserter(values));
REQUIRE(values == std::vector{1, 2, 3});
}
#endif
} // anonymous namespace
|
