1 //===-- stl_extras_test.cpp ------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file is a part of the ORC runtime.
10 //
11 // Note:
12 // This unit test was adapted from tests in
13 // llvm/unittests/ADT/STLExtrasTest.cpp
14 //
15 //===----------------------------------------------------------------------===//
16
17 #include "stl_extras.h"
18 #include "gtest/gtest.h"
19
20 using namespace __orc_rt;
21
TEST(STLExtrasTest,ApplyTuple)22 TEST(STLExtrasTest, ApplyTuple) {
23 auto T = std::make_tuple(1, 3, 7);
24 auto U = __orc_rt::apply_tuple(
25 [](int A, int B, int C) { return std::make_tuple(A - B, B - C, C - A); },
26 T);
27
28 EXPECT_EQ(-2, std::get<0>(U));
29 EXPECT_EQ(-4, std::get<1>(U));
30 EXPECT_EQ(6, std::get<2>(U));
31
32 auto V = __orc_rt::apply_tuple(
33 [](int A, int B, int C) {
34 return std::make_tuple(std::make_pair(A, char('A' + A)),
35 std::make_pair(B, char('A' + B)),
36 std::make_pair(C, char('A' + C)));
37 },
38 T);
39
40 EXPECT_EQ(std::make_pair(1, 'B'), std::get<0>(V));
41 EXPECT_EQ(std::make_pair(3, 'D'), std::get<1>(V));
42 EXPECT_EQ(std::make_pair(7, 'H'), std::get<2>(V));
43 }
44
45 class apply_variadic {
apply_one(int X)46 static int apply_one(int X) { return X + 1; }
apply_one(char C)47 static char apply_one(char C) { return C + 1; }
apply_one(std::string S)48 static std::string apply_one(std::string S) {
49 return S.substr(0, S.size() - 1);
50 }
51
52 public:
operator ()(Ts &&...Items)53 template <typename... Ts> auto operator()(Ts &&... Items) {
54 return std::make_tuple(apply_one(Items)...);
55 }
56 };
57
TEST(STLExtrasTest,ApplyTupleVariadic)58 TEST(STLExtrasTest, ApplyTupleVariadic) {
59 auto Items = std::make_tuple(1, std::string("Test"), 'X');
60 auto Values = __orc_rt::apply_tuple(apply_variadic(), Items);
61
62 EXPECT_EQ(2, std::get<0>(Values));
63 EXPECT_EQ("Tes", std::get<1>(Values));
64 EXPECT_EQ('Y', std::get<2>(Values));
65 }
66