blob: 4a21f452fa46763d4e63d937c19c8853f0c6b94d [file] [log] [blame]
Andrew Top61a84952019-04-30 15:07:33 -07001//===----------------------------------------------------------------------===//
2//
Kaido Kert788710a2023-06-05 07:50:22 -07003// 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
Andrew Top61a84952019-04-30 15:07:33 -07006//
7//===----------------------------------------------------------------------===//
8
Kaido Kert788710a2023-06-05 07:50:22 -07009// size_t count() const; // constexpr since C++23
Andrew Top61a84952019-04-30 15:07:33 -070010
11#include <bitset>
Andrew Top61a84952019-04-30 15:07:33 -070012#include <cassert>
Kaido Kert788710a2023-06-05 07:50:22 -070013#include <cstddef>
14#include <vector>
Andrew Top61a84952019-04-30 15:07:33 -070015
Kaido Kert788710a2023-06-05 07:50:22 -070016#include "../bitset_test_cases.h"
Andrew Top61a84952019-04-30 15:07:33 -070017#include "test_macros.h"
18
Kaido Kert788710a2023-06-05 07:50:22 -070019template <std::size_t N>
20TEST_CONSTEXPR_CXX23 void test_count() {
21 std::vector<std::bitset<N> > const cases = get_test_cases<N>();
22 for (std::size_t c = 0; c != cases.size(); ++c) {
23 const std::bitset<N> v = cases[c];
24 std::size_t c1 = v.count();
25 std::size_t c2 = 0;
26 for (std::size_t i = 0; i < v.size(); ++i)
27 if (v[i])
28 ++c2;
29 assert(c1 == c2);
30 }
31}
32
33TEST_CONSTEXPR_CXX23 bool test() {
34 test_count<0>();
35 test_count<1>();
36 test_count<31>();
37 test_count<32>();
38 test_count<33>();
39 test_count<63>();
40 test_count<64>();
41 test_count<65>();
42
43 return true;
44}
45
46int main(int, char**) {
47 test();
48 test_count<1000>(); // not in constexpr because of constexpr evaluation step limits
49#if TEST_STD_VER > 20
50 static_assert(test());
Andrew Top61a84952019-04-30 15:07:33 -070051#endif
52
Kaido Kert788710a2023-06-05 07:50:22 -070053 return 0;
Andrew Top61a84952019-04-30 15:07:33 -070054}