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 using namespace llvm;
  15
  16 namespace llvm {
  17
  18 TEST(ArrayRefTest, AllocatorCopy) {
  19   BumpPtrAllocator Alloc;
  20   static const uint16_t Words1[] = { 1, 4, 200, 37 };
  21   ArrayRef<uint16_t> Array1 = makeArrayRef(Words1, 4);
  22   static const uint16_t Words2[] = { 11, 4003, 67, 64000, 13 };
  23   ArrayRef<uint16_t> Array2 = makeArrayRef(Words2, 5);
  24   ArrayRef<uint16_t> Array1c = Array1.copy(Alloc);
  25   ArrayRef<uint16_t> Array2c = Array2.copy(Alloc);;
  26   EXPECT_TRUE(Array1.equals(Array1c));
  27   EXPECT_NE(Array1.data(), Array1c.data());
  28   EXPECT_TRUE(Array2.equals(Array2c));
  29   EXPECT_NE(Array2.data(), Array2c.data());
  30 }
  31
  32 TEST(ArrayRefTest, DropBack) {
  33   static const int TheNumbers[] = {4, 8, 15, 16, 23, 42};
  34   ArrayRef<int> AR1(TheNumbers);
  35   ArrayRef<int> AR2(TheNumbers, AR1.size() - 1);
  36   EXPECT_TRUE(AR1.drop_back().equals(AR2));
  37 }
  38
  39 TEST(ArrayRefTest, Equals) {
  40   static const int A1[] = {1, 2, 3, 4, 5, 6, 7, 8};
  41   ArrayRef<int> AR1(A1);
  42   EXPECT_TRUE(AR1.equals(1, 2, 3, 4, 5, 6, 7, 8));
  43   EXPECT_FALSE(AR1.equals(8, 1, 2, 4, 5, 6, 6, 7));
  44   EXPECT_FALSE(AR1.equals(2, 4, 5, 6, 6, 7, 8, 1));
  45   EXPECT_FALSE(AR1.equals(0, 1, 2, 4, 5, 6, 6, 7));
  46   EXPECT_FALSE(AR1.equals(1, 2, 42, 4, 5, 6, 7, 8));
  47   EXPECT_FALSE(AR1.equals(42, 2, 3, 4, 5, 6, 7, 8));
  48   EXPECT_FALSE(AR1.equals(1, 2, 3, 4, 5, 6, 7, 42));
  49   EXPECT_FALSE(AR1.equals(1, 2, 3, 4, 5, 6, 7));
  50   EXPECT_FALSE(AR1.equals(1, 2, 3, 4, 5, 6, 7, 8, 9));
  51
  52   ArrayRef<int> AR1a = AR1.drop_back();
  53   EXPECT_TRUE(AR1a.equals(1, 2, 3, 4, 5, 6, 7));
  54   EXPECT_FALSE(AR1a.equals(1, 2, 3, 4, 5, 6, 7, 8));
  55
  56   ArrayRef<int> AR1b = AR1a.slice(2, 4);
  57   EXPECT_TRUE(AR1b.equals(3, 4, 5, 6));
  58   EXPECT_FALSE(AR1b.equals(2, 3, 4, 5, 6));
  59   EXPECT_FALSE(AR1b.equals(3, 4, 5, 6, 7));
  60 }
  61
  62 } // end anonymous namespace