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