blob: 43e5005e62d8e1dbfb050fde38795e202e52485f [file] [log] [blame]
Nick Kledzik4d6d9812014-02-05 22:22:56 +00001//===- llvm/unittest/ADT/ArrayRefTest.cpp - ArrayRef unit tests -----------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#include "llvm/ADT/ArrayRef.h"
11#include "llvm/Support/Allocator.h"
12#include "llvm/Support/raw_ostream.h"
13#include "gtest/gtest.h"
Aaron Ballmana9614172015-02-17 17:44:07 +000014#include <vector>
Nick Kledzik4d6d9812014-02-05 22:22:56 +000015using namespace llvm;
16
David Blaikieee03ead2014-08-31 01:33:41 +000017// Check that the ArrayRef-of-pointer converting constructor only allows adding
18// cv qualifiers (not removing them, or otherwise changing the type)
19static_assert(
20 std::is_convertible<ArrayRef<int *>, ArrayRef<const int *>>::value,
21 "Adding const");
22static_assert(
23 std::is_convertible<ArrayRef<int *>, ArrayRef<volatile int *>>::value,
24 "Adding volatile");
25static_assert(!std::is_convertible<ArrayRef<int *>, ArrayRef<float *>>::value,
26 "Changing pointer of one type to a pointer of another");
27static_assert(
28 !std::is_convertible<ArrayRef<const int *>, ArrayRef<int *>>::value,
29 "Removing const");
30static_assert(
31 !std::is_convertible<ArrayRef<volatile int *>, ArrayRef<int *>>::value,
32 "Removing volatile");
33
Benjamin Kramer07f42cd2015-08-04 15:52:56 +000034namespace {
Nick Kledzik4d6d9812014-02-05 22:22:56 +000035
36TEST(ArrayRefTest, AllocatorCopy) {
37 BumpPtrAllocator Alloc;
38 static const uint16_t Words1[] = { 1, 4, 200, 37 };
39 ArrayRef<uint16_t> Array1 = makeArrayRef(Words1, 4);
40 static const uint16_t Words2[] = { 11, 4003, 67, 64000, 13 };
41 ArrayRef<uint16_t> Array2 = makeArrayRef(Words2, 5);
42 ArrayRef<uint16_t> Array1c = Array1.copy(Alloc);
Aaron Ballman3182ee92015-06-09 12:03:46 +000043 ArrayRef<uint16_t> Array2c = Array2.copy(Alloc);
Nick Kledzik4d6d9812014-02-05 22:22:56 +000044 EXPECT_TRUE(Array1.equals(Array1c));
45 EXPECT_NE(Array1.data(), Array1c.data());
46 EXPECT_TRUE(Array2.equals(Array2c));
47 EXPECT_NE(Array2.data(), Array2c.data());
Benjamin Kramer07f42cd2015-08-04 15:52:56 +000048
49 // Check that copy can cope with uninitialized memory.
50 struct NonAssignable {
51 const char *Ptr;
52
Benjamin Kramerbee48592015-08-05 09:39:41 +000053 NonAssignable(const char *Ptr) : Ptr(Ptr) {}
Benjamin Kramer07f42cd2015-08-04 15:52:56 +000054 NonAssignable(const NonAssignable &RHS) = default;
55 void operator=(const NonAssignable &RHS) { assert(RHS.Ptr != nullptr); }
56 bool operator==(const NonAssignable &RHS) const { return Ptr == RHS.Ptr; }
Benjamin Kramerbee48592015-08-05 09:39:41 +000057 } Array3Src[] = {"hello", "world"};
Benjamin Kramer07f42cd2015-08-04 15:52:56 +000058 ArrayRef<NonAssignable> Array3Copy = makeArrayRef(Array3Src).copy(Alloc);
59 EXPECT_EQ(makeArrayRef(Array3Src), Array3Copy);
60 EXPECT_NE(makeArrayRef(Array3Src).data(), Array3Copy.data());
Nick Kledzik4d6d9812014-02-05 22:22:56 +000061}
62
David Majnemer45647d82014-07-05 06:12:30 +000063TEST(ArrayRefTest, DropBack) {
64 static const int TheNumbers[] = {4, 8, 15, 16, 23, 42};
65 ArrayRef<int> AR1(TheNumbers);
66 ArrayRef<int> AR2(TheNumbers, AR1.size() - 1);
67 EXPECT_TRUE(AR1.drop_back().equals(AR2));
Ahmed Bougacha31a8a8e2016-06-02 17:26:03 +000068
69 // Check that drop_back accepts size_t-sized numbers.
70 ArrayRef<char> AR3((const char *)0x10000, SIZE_MAX - 0x10000);
71 EXPECT_EQ(1U, AR3.drop_back(AR3.size() - 1).size());
David Majnemer45647d82014-07-05 06:12:30 +000072}
73
Reid Kleckner1e4ba372016-05-03 20:53:20 +000074TEST(ArrayRefTest, DropFront) {
75 static const int TheNumbers[] = {4, 8, 15, 16, 23, 42};
76 ArrayRef<int> AR1(TheNumbers);
77 ArrayRef<int> AR2(&TheNumbers[2], AR1.size() - 2);
78 EXPECT_TRUE(AR1.drop_front(2).equals(AR2));
Ahmed Bougacha31a8a8e2016-06-02 17:26:03 +000079
80 // Check that drop_front accepts size_t-sized numbers.
81 ArrayRef<char> AR3((const char *)0x10000, SIZE_MAX - 0x10000);
82 EXPECT_EQ(1U, AR3.drop_front(AR3.size() - 1).size());
Reid Kleckner1e4ba372016-05-03 20:53:20 +000083}
84
Zachary Turner613c0752016-08-30 18:45:32 +000085TEST(ArrayRefTest, TakeBack) {
Zachary Turner84fc0592016-08-30 17:29:59 +000086 static const int TheNumbers[] = {4, 8, 15, 16, 23, 42};
87 ArrayRef<int> AR1(TheNumbers);
88 ArrayRef<int> AR2(AR1.end() - 1, 1);
Zachary Turner613c0752016-08-30 18:45:32 +000089 EXPECT_TRUE(AR1.take_back().equals(AR2));
Zachary Turner84fc0592016-08-30 17:29:59 +000090}
91
Zachary Turner613c0752016-08-30 18:45:32 +000092TEST(ArrayRefTest, TakeFront) {
Zachary Turner84fc0592016-08-30 17:29:59 +000093 static const int TheNumbers[] = {4, 8, 15, 16, 23, 42};
94 ArrayRef<int> AR1(TheNumbers);
95 ArrayRef<int> AR2(AR1.data(), 2);
Zachary Turner613c0752016-08-30 18:45:32 +000096 EXPECT_TRUE(AR1.take_front(2).equals(AR2));
Zachary Turner84fc0592016-08-30 17:29:59 +000097}
98
Chandler Carruth3ea985b2014-07-27 01:11:19 +000099TEST(ArrayRefTest, Equals) {
100 static const int A1[] = {1, 2, 3, 4, 5, 6, 7, 8};
101 ArrayRef<int> AR1(A1);
Benjamin Kramer030133c2015-03-01 21:05:05 +0000102 EXPECT_TRUE(AR1.equals({1, 2, 3, 4, 5, 6, 7, 8}));
103 EXPECT_FALSE(AR1.equals({8, 1, 2, 4, 5, 6, 6, 7}));
104 EXPECT_FALSE(AR1.equals({2, 4, 5, 6, 6, 7, 8, 1}));
105 EXPECT_FALSE(AR1.equals({0, 1, 2, 4, 5, 6, 6, 7}));
106 EXPECT_FALSE(AR1.equals({1, 2, 42, 4, 5, 6, 7, 8}));
107 EXPECT_FALSE(AR1.equals({42, 2, 3, 4, 5, 6, 7, 8}));
108 EXPECT_FALSE(AR1.equals({1, 2, 3, 4, 5, 6, 7, 42}));
109 EXPECT_FALSE(AR1.equals({1, 2, 3, 4, 5, 6, 7}));
110 EXPECT_FALSE(AR1.equals({1, 2, 3, 4, 5, 6, 7, 8, 9}));
Chandler Carruth3ea985b2014-07-27 01:11:19 +0000111
112 ArrayRef<int> AR1a = AR1.drop_back();
Benjamin Kramer030133c2015-03-01 21:05:05 +0000113 EXPECT_TRUE(AR1a.equals({1, 2, 3, 4, 5, 6, 7}));
114 EXPECT_FALSE(AR1a.equals({1, 2, 3, 4, 5, 6, 7, 8}));
Chandler Carruth3ea985b2014-07-27 01:11:19 +0000115
116 ArrayRef<int> AR1b = AR1a.slice(2, 4);
Benjamin Kramer030133c2015-03-01 21:05:05 +0000117 EXPECT_TRUE(AR1b.equals({3, 4, 5, 6}));
118 EXPECT_FALSE(AR1b.equals({2, 3, 4, 5, 6}));
119 EXPECT_FALSE(AR1b.equals({3, 4, 5, 6, 7}));
Chandler Carruth3ea985b2014-07-27 01:11:19 +0000120}
Nick Kledzik4d6d9812014-02-05 22:22:56 +0000121
Duncan P. N. Exon Smith349f6982014-08-19 19:18:46 +0000122TEST(ArrayRefTest, EmptyEquals) {
123 EXPECT_TRUE(ArrayRef<unsigned>() == ArrayRef<unsigned>());
124}
125
Ahmed Bougacha31a8a8e2016-06-02 17:26:03 +0000126TEST(ArrayRefTest, Slice) {
127 // Check that slice accepts size_t-sized numbers.
128 ArrayRef<char> AR((const char *)0x10000, SIZE_MAX - 0x10000);
129 EXPECT_EQ(1U, AR.slice(AR.size() - 1).size());
130 EXPECT_EQ(AR.size() - 1, AR.slice(1, AR.size() - 1).size());
131}
132
Craig Topperc9439cc2014-08-30 16:48:19 +0000133TEST(ArrayRefTest, ConstConvert) {
134 int buf[4];
135 for (int i = 0; i < 4; ++i)
136 buf[i] = i;
137
138 static int *A[] = {&buf[0], &buf[1], &buf[2], &buf[3]};
139 ArrayRef<const int *> a((ArrayRef<int *>(A)));
140 a = ArrayRef<int *>(A);
141}
142
Aaron Ballmana9614172015-02-17 17:44:07 +0000143static std::vector<int> ReturnTest12() { return {1, 2}; }
Aaron Ballmanfbb030f2015-02-17 17:08:08 +0000144static void ArgTest12(ArrayRef<int> A) {
Aaron Ballmana9614172015-02-17 17:44:07 +0000145 EXPECT_EQ(2U, A.size());
Aaron Ballmanfbb030f2015-02-17 17:08:08 +0000146 EXPECT_EQ(1, A[0]);
147 EXPECT_EQ(2, A[1]);
148}
149
Aaron Ballmanf4984a92015-02-17 15:37:53 +0000150TEST(ArrayRefTest, InitializerList) {
David Blaikie68ce7922016-08-25 22:09:13 +0000151 std::initializer_list<int> init_list = { 0, 1, 2, 3, 4 };
152 ArrayRef<int> A = init_list;
Aaron Ballmanf4984a92015-02-17 15:37:53 +0000153 for (int i = 0; i < 5; ++i)
154 EXPECT_EQ(i, A[i]);
Aaron Ballmanfbb030f2015-02-17 17:08:08 +0000155
Aaron Ballmana9614172015-02-17 17:44:07 +0000156 std::vector<int> B = ReturnTest12();
157 A = B;
Aaron Ballmanfbb030f2015-02-17 17:08:08 +0000158 EXPECT_EQ(1, A[0]);
159 EXPECT_EQ(2, A[1]);
160
161 ArgTest12({1, 2});
Aaron Ballmanf4984a92015-02-17 15:37:53 +0000162}
163
Mehdi Aminidefa5462015-09-10 00:05:04 +0000164// Test that makeArrayRef works on ArrayRef (no-op)
165TEST(ArrayRefTest, makeArrayRef) {
166 static const int A1[] = {1, 2, 3, 4, 5, 6, 7, 8};
167
168 // No copy expected for non-const ArrayRef (true no-op)
169 ArrayRef<int> AR1(A1);
170 ArrayRef<int> &AR1Ref = makeArrayRef(AR1);
171 EXPECT_EQ(&AR1, &AR1Ref);
172
173 // A copy is expected for non-const ArrayRef (thin copy)
174 const ArrayRef<int> AR2(A1);
175 const ArrayRef<int> &AR2Ref = makeArrayRef(AR2);
176 EXPECT_NE(&AR2Ref, &AR2);
177 EXPECT_TRUE(AR2.equals(AR2Ref));
178}
179
Nick Kledzik4d6d9812014-02-05 22:22:56 +0000180} // end anonymous namespace