unittests/ADT/ArrayRefTest.cpp

   1 //===- 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"
  14 #include <vector>
  15 using namespace llvm;
  16
  17 // Check that the ArrayRef-of-pointer converting constructor only allows adding
  18 // cv qualifiers (not removing them, or otherwise changing the type)
  19 static_assert(
  20     std::is_convertible<ArrayRef<int *>, ArrayRef<const int *>>::value,
  21     "Adding const");
  22 static_assert(
  23     std::is_convertible<ArrayRef<int *>, ArrayRef<volatile int *>>::value,
  24     "Adding volatile");
  25 static_assert(!std::is_convertible<ArrayRef<int *>, ArrayRef<float *>>::value,
  26               "Changing pointer of one type to a pointer of another");
  27 static_assert(
  28     !std::is_convertible<ArrayRef<const int *>, ArrayRef<int *>>::value,
  29     "Removing const");
  30 static_assert(
  31     !std::is_convertible<ArrayRef<volatile int *>, ArrayRef<int *>>::value,
  32     "Removing volatile");
  33
  34 namespace {
  35
  36 TEST(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);
  43   ArrayRef<uint16_t> Array2c = Array2.copy(Alloc);
  44   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());
  48
  49 #if __GNUC__==4 && __GNUC_MINOR__==8
  50   // g++-4.8 cannot compile the block below.
  51 #else
  52   // Check that copy can cope with uninitialized memory.
  53   struct NonAssignable {
  54     const char *Ptr;
  55
  56     NonAssignable(const NonAssignable &RHS) = default;
  57     void operator=(const NonAssignable &RHS) { assert(RHS.Ptr != nullptr); }
  58     bool operator==(const NonAssignable &RHS) const { return Ptr == RHS.Ptr; }
  59   } Array3Src[] = {{"hello"}, {"world"}};
  60   ArrayRef<NonAssignable> Array3Copy = makeArrayRef(Array3Src).copy(Alloc);
  61   EXPECT_EQ(makeArrayRef(Array3Src), Array3Copy);
  62   EXPECT_NE(makeArrayRef(Array3Src).data(), Array3Copy.data());
  63 #endif
  64 }
  65
  66 TEST(ArrayRefTest, DropBack) {
  67   static const int TheNumbers[] = {4, 8, 15, 16, 23, 42};
  68   ArrayRef<int> AR1(TheNumbers);
  69   ArrayRef<int> AR2(TheNumbers, AR1.size() - 1);
  70   EXPECT_TRUE(AR1.drop_back().equals(AR2));
  71 }
  72
  73 TEST(ArrayRefTest, Equals) {
  74   static const int A1[] = {1, 2, 3, 4, 5, 6, 7, 8};
  75   ArrayRef<int> AR1(A1);
  76   EXPECT_TRUE(AR1.equals({1, 2, 3, 4, 5, 6, 7, 8}));
  77   EXPECT_FALSE(AR1.equals({8, 1, 2, 4, 5, 6, 6, 7}));
  78   EXPECT_FALSE(AR1.equals({2, 4, 5, 6, 6, 7, 8, 1}));
  79   EXPECT_FALSE(AR1.equals({0, 1, 2, 4, 5, 6, 6, 7}));
  80   EXPECT_FALSE(AR1.equals({1, 2, 42, 4, 5, 6, 7, 8}));
  81   EXPECT_FALSE(AR1.equals({42, 2, 3, 4, 5, 6, 7, 8}));
  82   EXPECT_FALSE(AR1.equals({1, 2, 3, 4, 5, 6, 7, 42}));
  83   EXPECT_FALSE(AR1.equals({1, 2, 3, 4, 5, 6, 7}));
  84   EXPECT_FALSE(AR1.equals({1, 2, 3, 4, 5, 6, 7, 8, 9}));
  85
  86   ArrayRef<int> AR1a = AR1.drop_back();
  87   EXPECT_TRUE(AR1a.equals({1, 2, 3, 4, 5, 6, 7}));
  88   EXPECT_FALSE(AR1a.equals({1, 2, 3, 4, 5, 6, 7, 8}));
  89
  90   ArrayRef<int> AR1b = AR1a.slice(2, 4);
  91   EXPECT_TRUE(AR1b.equals({3, 4, 5, 6}));
  92   EXPECT_FALSE(AR1b.equals({2, 3, 4, 5, 6}));
  93   EXPECT_FALSE(AR1b.equals({3, 4, 5, 6, 7}));
  94 }
  95
  96 TEST(ArrayRefTest, EmptyEquals) {
  97   EXPECT_TRUE(ArrayRef<unsigned>() == ArrayRef<unsigned>());
  98 }
  99
 100 TEST(ArrayRefTest, ConstConvert) {
 101   int buf[4];
 102   for (int i = 0; i < 4; ++i)
 103     buf[i] = i;
 104
 105   static int *A[] = {&buf[0], &buf[1], &buf[2], &buf[3]};
 106   ArrayRef<const int *> a((ArrayRef<int *>(A)));
 107   a = ArrayRef<int *>(A);
 108 }
 109
 110 static std::vector<int> ReturnTest12() { return {1, 2}; }
 111 static void ArgTest12(ArrayRef<int> A) {
 112   EXPECT_EQ(2U, A.size());
 113   EXPECT_EQ(1, A[0]);
 114   EXPECT_EQ(2, A[1]);
 115 }
 116
 117 TEST(ArrayRefTest, InitializerList) {
 118   ArrayRef<int> A = { 0, 1, 2, 3, 4 };
 119   for (int i = 0; i < 5; ++i)
 120     EXPECT_EQ(i, A[i]);
 121
 122   std::vector<int> B = ReturnTest12();
 123   A = B;
 124   EXPECT_EQ(1, A[0]);
 125   EXPECT_EQ(2, A[1]);
 126
 127   ArgTest12({1, 2});
 128 }
 129
 130 } // end anonymous namespace