right_shift_eq.pass.cpp revision b64f8b07c104c6cc986570ac8ee0ed16a9f23976 
1//===----------------------------------------------------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is dual licensed under the MIT and the University of Illinois Open
6// Source Licenses. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10// test bitset<N>& operator<<=(size_t pos);
11
12#include <bitset>
13#include <cstdlib>
14#include <cassert>
15
16template <std::size_t N>
17std::bitset<N>
18make_bitset()
19{
20    std::bitset<N> v;
21    for (std::size_t i = 0; i < N; ++i)
22        v[i] = static_cast<bool>(std::rand() & 1);
23    return v;
24}
25
26template <std::size_t N>
27void test_right_shift()
28{
29    for (std::size_t s = 0; s <= N+1; ++s)
30    {
31        std::bitset<N> v1 = make_bitset<N>();
32        std::bitset<N> v2 = v1;
33        v1 >>= s;
34        for (std::size_t i = 0; i < N; ++i)
35            if (i + s < N)
36                assert(v1[i] == v2[i + s]);
37            else
38                assert(v1[i] == 0);
39    }
40}
41
42int main()
43{
44    test_right_shift<0>();
45    test_right_shift<1>();
46    test_right_shift<31>();
47    test_right_shift<32>();
48    test_right_shift<33>();
49    test_right_shift<63>();
50    test_right_shift<64>();
51    test_right_shift<65>();
52    test_right_shift<1000>();
53}
54