libcxx/test/std/utilities/template.bitset/bitset.members/right_shift_eq.pass.cpp

   1 //===----------------------------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 // test bitset<N>& operator<<=(size_t pos);
  10
  11 #include <bitset>
  12 #include <cassert>
  13 #include <cstddef>
  14 #include <vector>
  15
  16 #include "../bitset_test_cases.h"
  17 #include "test_macros.h"
  18
  19 template <std::size_t N>
  20 void test_right_shift() {
  21     std::vector<std::bitset<N> > const cases = get_test_cases<N>();
  22     for (std::size_t c = 0; c != cases.size(); ++c) {
  23         for (std::size_t s = 0; s <= N+1; ++s) {
  24             std::bitset<N> v1 = cases[c];
  25             std::bitset<N> v2 = v1;
  26             v1 >>= s;
  27             for (std::size_t i = 0; i < v1.size(); ++i)
  28                 if (i + s < v1.size())
  29                     assert(v1[i] == v2[i + s]);
  30                 else
  31                     assert(v1[i] == 0);
  32         }
  33     }
  34 }
  35
  36 int main(int, char**) {
  37     test_right_shift<0>();
  38     test_right_shift<1>();
  39     test_right_shift<31>();
  40     test_right_shift<32>();
  41     test_right_shift<33>();
  42     test_right_shift<63>();
  43     test_right_shift<64>();
  44     test_right_shift<65>();
  45     test_right_shift<1000>();
  46
  47     return 0;
  48 }