unittests/ADT/APIntTest.cpp

   1 //===- llvm/unittest/ADT/APInt.cpp - APInt 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 <ostream>
  11 #include "llvm/Support/raw_ostream.h"
  12 #include "gtest/gtest.h"
  13 #include "llvm/ADT/APInt.h"
  14 #include "llvm/ADT/SmallString.h"
  15
  16 using namespace llvm;
  17
  18 namespace {
  19
  20 // Test that APInt shift left works when bitwidth > 64 and shiftamt == 0
  21 TEST(APIntTest, ShiftLeftByZero) {
  22   APInt One = APInt::getNullValue(65) + 1;
  23   APInt Shl = One.shl(0);
  24   EXPECT_EQ(true, Shl[0]);
  25   EXPECT_EQ(false, Shl[1]);
  26 }
  27
  28 TEST(APIntTest, i128_NegativeCount) {
  29   APInt Minus3(128, static_cast<uint64_t>(-3), true);
  30   EXPECT_EQ(126u, Minus3.countLeadingOnes());
  31   EXPECT_EQ(-3, Minus3.getSExtValue());
  32
  33   APInt Minus1(128, static_cast<uint64_t>(-1), true);
  34   EXPECT_EQ(0u, Minus1.countLeadingZeros());
  35   EXPECT_EQ(128u, Minus1.countLeadingOnes());
  36   EXPECT_EQ(128u, Minus1.getActiveBits());
  37   EXPECT_EQ(0u, Minus1.countTrailingZeros());
  38   EXPECT_EQ(128u, Minus1.countTrailingOnes());
  39   EXPECT_EQ(128u, Minus1.countPopulation());
  40   EXPECT_EQ(-1, Minus1.getSExtValue());
  41 }
  42
  43 TEST(APIntTest, i33_Count) {
  44   APInt i33minus2(33, static_cast<uint64_t>(-2), true);
  45   EXPECT_EQ(0u, i33minus2.countLeadingZeros());
  46   EXPECT_EQ(32u, i33minus2.countLeadingOnes());
  47   EXPECT_EQ(33u, i33minus2.getActiveBits());
  48   EXPECT_EQ(1u, i33minus2.countTrailingZeros());
  49   EXPECT_EQ(32u, i33minus2.countPopulation());
  50   EXPECT_EQ(-2, i33minus2.getSExtValue());
  51   EXPECT_EQ(((uint64_t)-2)&((1ull<<33) -1), i33minus2.getZExtValue());
  52 }
  53
  54 TEST(APIntTest, i65_Count) {
  55   APInt i65minus(65, 0, true);
  56   i65minus.set(64);
  57   EXPECT_EQ(0u, i65minus.countLeadingZeros());
  58   EXPECT_EQ(1u, i65minus.countLeadingOnes());
  59   EXPECT_EQ(65u, i65minus.getActiveBits());
  60   EXPECT_EQ(64u, i65minus.countTrailingZeros());
  61   EXPECT_EQ(1u, i65minus.countPopulation());
  62 }
  63
  64 TEST(APIntTest, i128_PositiveCount) {
  65   APInt u128max = APInt::getAllOnesValue(128);
  66   EXPECT_EQ(128u, u128max.countLeadingOnes());
  67   EXPECT_EQ(0u, u128max.countLeadingZeros());
  68   EXPECT_EQ(128u, u128max.getActiveBits());
  69   EXPECT_EQ(0u, u128max.countTrailingZeros());
  70   EXPECT_EQ(128u, u128max.countTrailingOnes());
  71   EXPECT_EQ(128u, u128max.countPopulation());
  72
  73   APInt u64max(128, static_cast<uint64_t>(-1), false);
  74   EXPECT_EQ(64u, u64max.countLeadingZeros());
  75   EXPECT_EQ(0u, u64max.countLeadingOnes());
  76   EXPECT_EQ(64u, u64max.getActiveBits());
  77   EXPECT_EQ(0u, u64max.countTrailingZeros());
  78   EXPECT_EQ(64u, u64max.countTrailingOnes());
  79   EXPECT_EQ(64u, u64max.countPopulation());
  80   EXPECT_EQ((uint64_t)~0ull, u64max.getZExtValue());
  81
  82   APInt zero(128, 0, true);
  83   EXPECT_EQ(128u, zero.countLeadingZeros());
  84   EXPECT_EQ(0u, zero.countLeadingOnes());
  85   EXPECT_EQ(0u, zero.getActiveBits());
  86   EXPECT_EQ(128u, zero.countTrailingZeros());
  87   EXPECT_EQ(0u, zero.countTrailingOnes());
  88   EXPECT_EQ(0u, zero.countPopulation());
  89   EXPECT_EQ(0u, zero.getSExtValue());
  90   EXPECT_EQ(0u, zero.getZExtValue());
  91
  92   APInt one(128, 1, true);
  93   EXPECT_EQ(127u, one.countLeadingZeros());
  94   EXPECT_EQ(0u, one.countLeadingOnes());
  95   EXPECT_EQ(1u, one.getActiveBits());
  96   EXPECT_EQ(0u, one.countTrailingZeros());
  97   EXPECT_EQ(1u, one.countTrailingOnes());
  98   EXPECT_EQ(1u, one.countPopulation());
  99   EXPECT_EQ(1, one.getSExtValue());
 100   EXPECT_EQ(1u, one.getZExtValue());
 101 }
 102
 103 TEST(APIntTest, i1) {
 104   const APInt neg_two(1, static_cast<uint64_t>(-2), true);
 105   const APInt neg_one(1, static_cast<uint64_t>(-1), true);
 106   const APInt zero(1, 0);
 107   const APInt one(1, 1);
 108   const APInt two(1, 2);
 109
 110   EXPECT_EQ(0, neg_two.getSExtValue());
 111   EXPECT_EQ(-1, neg_one.getSExtValue());
 112   EXPECT_EQ(1u, neg_one.getZExtValue());
 113   EXPECT_EQ(0u, zero.getZExtValue());
 114   EXPECT_EQ(-1, one.getSExtValue());
 115   EXPECT_EQ(1u, one.getZExtValue());
 116   EXPECT_EQ(0u, two.getZExtValue());
 117   EXPECT_EQ(0, two.getSExtValue());
 118
 119   // Basic equalities for 1-bit values.
 120   EXPECT_EQ(zero, two);
 121   EXPECT_EQ(zero, neg_two);
 122   EXPECT_EQ(one, neg_one);
 123   EXPECT_EQ(two, neg_two);
 124
 125   // Additions.
 126   EXPECT_EQ(two, one + one);
 127   EXPECT_EQ(zero, neg_one + one);
 128   EXPECT_EQ(neg_two, neg_one + neg_one);
 129
 130   // Subtractions.
 131   EXPECT_EQ(neg_two, neg_one - one);
 132   EXPECT_EQ(two, one - neg_one);
 133   EXPECT_EQ(zero, one - one);
 134
 135   // Shifts.
 136   EXPECT_EQ(zero, one << one);
 137   EXPECT_EQ(one, one << zero);
 138   EXPECT_EQ(zero, one.shl(1));
 139   EXPECT_EQ(one, one.shl(0));
 140   EXPECT_EQ(zero, one.lshr(1));
 141   EXPECT_EQ(zero, one.ashr(1));
 142
 143   // Multiplies.
 144   EXPECT_EQ(neg_one, neg_one * one);
 145   EXPECT_EQ(neg_one, one * neg_one);
 146   EXPECT_EQ(one, neg_one * neg_one);
 147   EXPECT_EQ(one, one * one);
 148
 149   // Divides.
 150   EXPECT_EQ(neg_one, one.sdiv(neg_one));
 151   EXPECT_EQ(neg_one, neg_one.sdiv(one));
 152   EXPECT_EQ(one, neg_one.sdiv(neg_one));
 153   EXPECT_EQ(one, one.sdiv(one));
 154
 155   EXPECT_EQ(neg_one, one.udiv(neg_one));
 156   EXPECT_EQ(neg_one, neg_one.udiv(one));
 157   EXPECT_EQ(one, neg_one.udiv(neg_one));
 158   EXPECT_EQ(one, one.udiv(one));
 159
 160   // Remainders.
 161   EXPECT_EQ(zero, neg_one.srem(one));
 162   EXPECT_EQ(zero, neg_one.urem(one));
 163   EXPECT_EQ(zero, one.srem(neg_one));
 164 }
 165
 166 TEST(APIntTest, fromString) {
 167   EXPECT_EQ(APInt(32, 0), APInt(32,   "0", 2));
 168   EXPECT_EQ(APInt(32, 1), APInt(32,   "1", 2));
 169   EXPECT_EQ(APInt(32, 2), APInt(32,  "10", 2));
 170   EXPECT_EQ(APInt(32, 3), APInt(32,  "11", 2));
 171   EXPECT_EQ(APInt(32, 4), APInt(32, "100", 2));
 172
 173   EXPECT_EQ(APInt(32, 0), APInt(32,   "+0", 2));
 174   EXPECT_EQ(APInt(32, 1), APInt(32,   "+1", 2));
 175   EXPECT_EQ(APInt(32, 2), APInt(32,  "+10", 2));
 176   EXPECT_EQ(APInt(32, 3), APInt(32,  "+11", 2));
 177   EXPECT_EQ(APInt(32, 4), APInt(32, "+100", 2));
 178
 179   EXPECT_EQ(APInt(32, -0), APInt(32,   "-0", 2));
 180   EXPECT_EQ(APInt(32, -1), APInt(32,   "-1", 2));
 181   EXPECT_EQ(APInt(32, -2), APInt(32,  "-10", 2));
 182   EXPECT_EQ(APInt(32, -3), APInt(32,  "-11", 2));
 183   EXPECT_EQ(APInt(32, -4), APInt(32, "-100", 2));
 184
 185
 186   EXPECT_EQ(APInt(32,  0), APInt(32,  "0",  8));
 187   EXPECT_EQ(APInt(32,  1), APInt(32,  "1",  8));
 188   EXPECT_EQ(APInt(32,  7), APInt(32,  "7",  8));
 189   EXPECT_EQ(APInt(32,  8), APInt(32,  "10", 8));
 190   EXPECT_EQ(APInt(32, 15), APInt(32,  "17", 8));
 191   EXPECT_EQ(APInt(32, 16), APInt(32,  "20", 8));
 192
 193   EXPECT_EQ(APInt(32,  +0), APInt(32,  "+0",  8));
 194   EXPECT_EQ(APInt(32,  +1), APInt(32,  "+1",  8));
 195   EXPECT_EQ(APInt(32,  +7), APInt(32,  "+7",  8));
 196   EXPECT_EQ(APInt(32,  +8), APInt(32,  "+10", 8));
 197   EXPECT_EQ(APInt(32, +15), APInt(32,  "+17", 8));
 198   EXPECT_EQ(APInt(32, +16), APInt(32,  "+20", 8));
 199
 200   EXPECT_EQ(APInt(32,  -0), APInt(32,  "-0",  8));
 201   EXPECT_EQ(APInt(32,  -1), APInt(32,  "-1",  8));
 202   EXPECT_EQ(APInt(32,  -7), APInt(32,  "-7",  8));
 203   EXPECT_EQ(APInt(32,  -8), APInt(32,  "-10", 8));
 204   EXPECT_EQ(APInt(32, -15), APInt(32,  "-17", 8));
 205   EXPECT_EQ(APInt(32, -16), APInt(32,  "-20", 8));
 206
 207
 208   EXPECT_EQ(APInt(32,  0), APInt(32,  "0", 10));
 209   EXPECT_EQ(APInt(32,  1), APInt(32,  "1", 10));
 210   EXPECT_EQ(APInt(32,  9), APInt(32,  "9", 10));
 211   EXPECT_EQ(APInt(32, 10), APInt(32, "10", 10));
 212   EXPECT_EQ(APInt(32, 19), APInt(32, "19", 10));
 213   EXPECT_EQ(APInt(32, 20), APInt(32, "20", 10));
 214
 215   EXPECT_EQ(APInt(32,  -0), APInt(32,  "-0", 10));
 216   EXPECT_EQ(APInt(32,  -1), APInt(32,  "-1", 10));
 217   EXPECT_EQ(APInt(32,  -9), APInt(32,  "-9", 10));
 218   EXPECT_EQ(APInt(32, -10), APInt(32, "-10", 10));
 219   EXPECT_EQ(APInt(32, -19), APInt(32, "-19", 10));
 220   EXPECT_EQ(APInt(32, -20), APInt(32, "-20", 10));
 221
 222
 223   EXPECT_EQ(APInt(32,  0), APInt(32,  "0", 16));
 224   EXPECT_EQ(APInt(32,  1), APInt(32,  "1", 16));
 225   EXPECT_EQ(APInt(32, 15), APInt(32,  "F", 16));
 226   EXPECT_EQ(APInt(32, 16), APInt(32, "10", 16));
 227   EXPECT_EQ(APInt(32, 31), APInt(32, "1F", 16));
 228   EXPECT_EQ(APInt(32, 32), APInt(32, "20", 16));
 229
 230   EXPECT_EQ(APInt(32,  -0), APInt(32,  "-0", 16));
 231   EXPECT_EQ(APInt(32,  -1), APInt(32,  "-1", 16));
 232   EXPECT_EQ(APInt(32, -15), APInt(32,  "-F", 16));
 233   EXPECT_EQ(APInt(32, -16), APInt(32, "-10", 16));
 234   EXPECT_EQ(APInt(32, -31), APInt(32, "-1F", 16));
 235   EXPECT_EQ(APInt(32, -32), APInt(32, "-20", 16));
 236 }
 237
 238 TEST(APIntTest, StringBitsNeeded2) {
 239   EXPECT_EQ(1U, APInt::getBitsNeeded(  "0", 2));
 240   EXPECT_EQ(1U, APInt::getBitsNeeded(  "1", 2));
 241   EXPECT_EQ(2U, APInt::getBitsNeeded( "10", 2));
 242   EXPECT_EQ(2U, APInt::getBitsNeeded( "11", 2));
 243   EXPECT_EQ(3U, APInt::getBitsNeeded("100", 2));
 244
 245   EXPECT_EQ(1U, APInt::getBitsNeeded(  "+0", 2));
 246   EXPECT_EQ(1U, APInt::getBitsNeeded(  "+1", 2));
 247   EXPECT_EQ(2U, APInt::getBitsNeeded( "+10", 2));
 248   EXPECT_EQ(2U, APInt::getBitsNeeded( "+11", 2));
 249   EXPECT_EQ(3U, APInt::getBitsNeeded("+100", 2));
 250
 251   EXPECT_EQ(2U, APInt::getBitsNeeded(  "-0", 2));
 252   EXPECT_EQ(2U, APInt::getBitsNeeded(  "-1", 2));
 253   EXPECT_EQ(3U, APInt::getBitsNeeded( "-10", 2));
 254   EXPECT_EQ(3U, APInt::getBitsNeeded( "-11", 2));
 255   EXPECT_EQ(4U, APInt::getBitsNeeded("-100", 2));
 256 }
 257
 258 TEST(APIntTest, StringBitsNeeded8) {
 259   EXPECT_EQ(3U, APInt::getBitsNeeded( "0", 8));
 260   EXPECT_EQ(3U, APInt::getBitsNeeded( "7", 8));
 261   EXPECT_EQ(6U, APInt::getBitsNeeded("10", 8));
 262   EXPECT_EQ(6U, APInt::getBitsNeeded("17", 8));
 263   EXPECT_EQ(6U, APInt::getBitsNeeded("20", 8));
 264
 265   EXPECT_EQ(3U, APInt::getBitsNeeded( "+0", 8));
 266   EXPECT_EQ(3U, APInt::getBitsNeeded( "+7", 8));
 267   EXPECT_EQ(6U, APInt::getBitsNeeded("+10", 8));
 268   EXPECT_EQ(6U, APInt::getBitsNeeded("+17", 8));
 269   EXPECT_EQ(6U, APInt::getBitsNeeded("+20", 8));
 270
 271   EXPECT_EQ(4U, APInt::getBitsNeeded( "-0", 8));
 272   EXPECT_EQ(4U, APInt::getBitsNeeded( "-7", 8));
 273   EXPECT_EQ(7U, APInt::getBitsNeeded("-10", 8));
 274   EXPECT_EQ(7U, APInt::getBitsNeeded("-17", 8));
 275   EXPECT_EQ(7U, APInt::getBitsNeeded("-20", 8));
 276 }
 277
 278 TEST(APIntTest, StringBitsNeeded10) {
 279   EXPECT_EQ(1U, APInt::getBitsNeeded( "0", 10));
 280   EXPECT_EQ(2U, APInt::getBitsNeeded( "3", 10));
 281   EXPECT_EQ(4U, APInt::getBitsNeeded( "9", 10));
 282   EXPECT_EQ(4U, APInt::getBitsNeeded("10", 10));
 283   EXPECT_EQ(5U, APInt::getBitsNeeded("19", 10));
 284   EXPECT_EQ(5U, APInt::getBitsNeeded("20", 10));
 285
 286   EXPECT_EQ(1U, APInt::getBitsNeeded( "+0", 10));
 287   EXPECT_EQ(4U, APInt::getBitsNeeded( "+9", 10));
 288   EXPECT_EQ(4U, APInt::getBitsNeeded("+10", 10));
 289   EXPECT_EQ(5U, APInt::getBitsNeeded("+19", 10));
 290   EXPECT_EQ(5U, APInt::getBitsNeeded("+20", 10));
 291
 292   EXPECT_EQ(2U, APInt::getBitsNeeded( "-0", 10));
 293   EXPECT_EQ(5U, APInt::getBitsNeeded( "-9", 10));
 294   EXPECT_EQ(5U, APInt::getBitsNeeded("-10", 10));
 295   EXPECT_EQ(6U, APInt::getBitsNeeded("-19", 10));
 296   EXPECT_EQ(6U, APInt::getBitsNeeded("-20", 10));
 297 }
 298
 299 TEST(APIntTest, StringBitsNeeded16) {
 300   EXPECT_EQ(4U, APInt::getBitsNeeded( "0", 16));
 301   EXPECT_EQ(4U, APInt::getBitsNeeded( "F", 16));
 302   EXPECT_EQ(8U, APInt::getBitsNeeded("10", 16));
 303   EXPECT_EQ(8U, APInt::getBitsNeeded("1F", 16));
 304   EXPECT_EQ(8U, APInt::getBitsNeeded("20", 16));
 305
 306   EXPECT_EQ(4U, APInt::getBitsNeeded( "+0", 16));
 307   EXPECT_EQ(4U, APInt::getBitsNeeded( "+F", 16));
 308   EXPECT_EQ(8U, APInt::getBitsNeeded("+10", 16));
 309   EXPECT_EQ(8U, APInt::getBitsNeeded("+1F", 16));
 310   EXPECT_EQ(8U, APInt::getBitsNeeded("+20", 16));
 311
 312   EXPECT_EQ(5U, APInt::getBitsNeeded( "-0", 16));
 313   EXPECT_EQ(5U, APInt::getBitsNeeded( "-F", 16));
 314   EXPECT_EQ(9U, APInt::getBitsNeeded("-10", 16));
 315   EXPECT_EQ(9U, APInt::getBitsNeeded("-1F", 16));
 316   EXPECT_EQ(9U, APInt::getBitsNeeded("-20", 16));
 317 }
 318
 319
 320 #ifdef GTEST_HAS_DEATH_TEST
 321 TEST(APIntTest, StringDeath) {
 322   EXPECT_DEATH(APInt(0, "", 0), "Bitwidth too small");
 323   EXPECT_DEATH(APInt(32, "", 0), "Invalid string length");
 324   EXPECT_DEATH(APInt(32, "0", 0), "Radix should be 2, 8, 10, or 16!");
 325   EXPECT_DEATH(APInt(32, "", 10), "Invalid string length");
 326   EXPECT_DEATH(APInt(32, "-", 10), "String is only a sign, needs a value.");
 327   EXPECT_DEATH(APInt(1, "1234", 10), "Insufficient bit width");
 328   EXPECT_DEATH(APInt(32, "\0", 10), "Invalid string length");
 329   EXPECT_DEATH(APInt(32, StringRef("1\02", 3), 10), "Invalid character in digit string");
 330   EXPECT_DEATH(APInt(32, "1L", 10), "Invalid character in digit string");
 331 }
 332 #endif
 333
 334 }