1 // Copyright 2021 The Tint Authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "src/utils/transform.h"
16
17 #include <string>
18 #include <type_traits>
19
20 #include "gmock/gmock.h"
21
22 #define CHECK_ELEMENT_TYPE(vector, expected) \
23 static_assert(std::is_same<decltype(vector)::value_type, expected>::value, \
24 "unexpected result vector element type")
25
26 namespace tint {
27 namespace utils {
28 namespace {
29
TEST(TransformTest,Empty)30 TEST(TransformTest, Empty) {
31 const std::vector<int> empty{};
32 {
33 auto transformed = Transform(empty, [](int) -> int {
34 [] { FAIL() << "Transform should not be called for empty vector"; }();
35 return 0;
36 });
37 CHECK_ELEMENT_TYPE(transformed, int);
38 EXPECT_EQ(transformed.size(), 0u);
39 }
40 {
41 auto transformed = Transform(empty, [](int, size_t) -> int {
42 [] { FAIL() << "Transform should not be called for empty vector"; }();
43 return 0;
44 });
45 CHECK_ELEMENT_TYPE(transformed, int);
46 EXPECT_EQ(transformed.size(), 0u);
47 }
48 }
49
TEST(TransformTest,Identity)50 TEST(TransformTest, Identity) {
51 const std::vector<int> input{1, 2, 3, 4};
52 {
53 auto transformed = Transform(input, [](int i) { return i; });
54 CHECK_ELEMENT_TYPE(transformed, int);
55 EXPECT_THAT(transformed, testing::ElementsAre(1, 2, 3, 4));
56 }
57 {
58 auto transformed = Transform(input, [](int i, size_t) { return i; });
59 CHECK_ELEMENT_TYPE(transformed, int);
60 EXPECT_THAT(transformed, testing::ElementsAre(1, 2, 3, 4));
61 }
62 }
63
TEST(TransformTest,Index)64 TEST(TransformTest, Index) {
65 const std::vector<int> input{10, 20, 30, 40};
66 {
67 auto transformed = Transform(input, [](int, size_t idx) { return idx; });
68 CHECK_ELEMENT_TYPE(transformed, size_t);
69 EXPECT_THAT(transformed, testing::ElementsAre(0u, 1u, 2u, 3u));
70 }
71 }
72
TEST(TransformTest,TransformSameType)73 TEST(TransformTest, TransformSameType) {
74 const std::vector<int> input{1, 2, 3, 4};
75 {
76 auto transformed = Transform(input, [](int i) { return i * 10; });
77 CHECK_ELEMENT_TYPE(transformed, int);
78 EXPECT_THAT(transformed, testing::ElementsAre(10, 20, 30, 40));
79 }
80 }
81
TEST(TransformTest,TransformDifferentType)82 TEST(TransformTest, TransformDifferentType) {
83 const std::vector<int> input{1, 2, 3, 4};
84 {
85 auto transformed =
86 Transform(input, [](int i) { return std::to_string(i); });
87 CHECK_ELEMENT_TYPE(transformed, std::string);
88 EXPECT_THAT(transformed, testing::ElementsAre("1", "2", "3", "4"));
89 }
90 }
91
92 } // namespace
93 } // namespace utils
94 } // namespace tint
95