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
119
120
121
122
123
124
125
|
//===--- TransformRangeTest.cpp -------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#include "swift/Basic/STLExtras.h"
#include "llvm/ADT/ArrayRef.h"
#include "gtest/gtest.h"
using namespace swift;
TEST(TransformRangeTest, Empty) {
auto transform = [](int i) -> float { return float(i); };
std::function<float (int)> f(transform);
std::vector<int> v1;
auto EmptyArray = makeTransformRange(llvm::ArrayRef<int>(v1), f);
EXPECT_EQ(EmptyArray.empty(), v1.empty());
}
TEST(TransformRangeTest, Subscript) {
auto transform = [](int i) -> float { return float(i); };
std::function<float (int)> f(transform);
std::vector<int> v1;
v1.push_back(0);
v1.push_back(2);
v1.push_back(3);
v1.push_back(100);
v1.push_back(-5);
v1.push_back(-30);
auto Array = makeTransformRange(llvm::ArrayRef<int>(v1), f);
EXPECT_EQ(Array.size(), v1.size());
for (unsigned i = 0, e = Array.size(); i != e; ++i) {
EXPECT_EQ(Array[i], transform(v1[i]));
}
}
TEST(TransformRangeTest, Iteration) {
auto transform = [](int i) -> float { return float(i); };
std::function<float (int)> f(transform);
std::vector<int> v1;
v1.push_back(0);
v1.push_back(2);
v1.push_back(3);
v1.push_back(100);
v1.push_back(-5);
v1.push_back(-30);
auto Array = makeTransformRange(llvm::ArrayRef<int>(v1), f);
auto VBegin = v1.begin();
auto VIter = v1.begin();
auto VEnd = v1.end();
auto TBegin = Array.begin();
auto TIter = Array.begin();
auto TEnd = Array.end();
// Forwards.
while (VIter != VEnd) {
EXPECT_NE(TIter, TEnd);
EXPECT_EQ(transform(*VIter), *TIter);
++VIter;
++TIter;
}
// Backwards.
while (VIter != VBegin) {
EXPECT_NE(TIter, TBegin);
--VIter;
--TIter;
EXPECT_EQ(transform(*VIter), *TIter);
}
}
TEST(TransformRangeTest, IterationWithSizelessSubscriptlessRange) {
auto transform = [](int i) -> float { return float(i); };
std::function<float (int)> f(transform);
std::vector<int> v1;
v1.push_back(0);
v1.push_back(2);
v1.push_back(3);
v1.push_back(100);
v1.push_back(-5);
v1.push_back(-30);
auto Array = makeTransformRange(llvm::make_range(v1.begin(), v1.end()), f);
auto VBegin = v1.begin();
auto VIter = v1.begin();
auto VEnd = v1.end();
auto TBegin = Array.begin();
auto TIter = Array.begin();
auto TEnd = Array.end();
// Forwards.
while (VIter != VEnd) {
EXPECT_NE(TIter, TEnd);
EXPECT_EQ(transform(*VIter), *TIter);
++VIter;
++TIter;
}
// Backwards.
while (VIter != VBegin) {
EXPECT_NE(TIter, TBegin);
--VIter;
--TIter;
EXPECT_EQ(transform(*VIter), *TIter);
}
}
|
