114 lines
3 KiB
C++
114 lines
3 KiB
C++
#define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
|
|
|
|
#include <algorithm>
|
|
#include <random>
|
|
|
|
#include "doctest/doctest.h"
|
|
#include "test_common.hpp"
|
|
#include "xcdat/bit_vector.hpp"
|
|
|
|
std::uint64_t get_num_ones(const std::vector<bool>& bits) {
|
|
return std::accumulate(bits.begin(), bits.end(), 0ULL);
|
|
}
|
|
|
|
std::uint64_t rank_naive(const std::vector<bool>& bits, std::uint64_t i) {
|
|
return std::accumulate(bits.begin(), bits.begin() + i, 0ULL);
|
|
}
|
|
|
|
std::uint64_t select_naive(const std::vector<bool>& bits, std::uint64_t n) {
|
|
std::uint64_t i = 0;
|
|
for (; i < bits.size(); i++) {
|
|
if (bits[i]) {
|
|
if (n == 0) {
|
|
break;
|
|
}
|
|
n -= 1;
|
|
}
|
|
}
|
|
return i;
|
|
}
|
|
|
|
void test_rank_select(const std::vector<bool>& bits) {
|
|
xcdat::bit_vector bv;
|
|
{
|
|
xcdat::bit_vector::builder bvb(bits.size());
|
|
for (std::uint64_t i = 0; i < bits.size(); i++) {
|
|
bvb.set_bit(i, bits[i]);
|
|
}
|
|
bv.build(bvb, true, true);
|
|
}
|
|
|
|
REQUIRE_EQ(bv.size(), bits.size());
|
|
REQUIRE_EQ(bv.num_ones(), get_num_ones(bits));
|
|
|
|
for (std::uint64_t i = 0; i < bits.size(); i++) {
|
|
REQUIRE_EQ(bv[i], bits[i]);
|
|
}
|
|
|
|
static constexpr std::uint64_t seed = 17;
|
|
std::mt19937_64 engine(seed);
|
|
|
|
{
|
|
std::uniform_int_distribution<std::uint64_t> dist(0, bv.size());
|
|
for (std::uint64_t r = 0; r < 100; r++) {
|
|
const std::uint64_t i = dist(engine);
|
|
REQUIRE_EQ(bv.rank(i), rank_naive(bits, i));
|
|
}
|
|
}
|
|
if (bv.num_ones() != 0) {
|
|
std::uniform_int_distribution<std::uint64_t> dist(0, bv.num_ones() - 1);
|
|
for (std::uint64_t r = 0; r < 100; r++) {
|
|
const std::uint64_t n = dist(engine);
|
|
REQUIRE_EQ(bv.select(n), select_naive(bits, n));
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Test bit_vector::builder with resize") {
|
|
const auto bits = xcdat::test::make_random_bits(10000);
|
|
|
|
xcdat::bit_vector::builder bvb;
|
|
bvb.resize(bits.size());
|
|
|
|
REQUIRE_EQ(bvb.size(), bits.size());
|
|
|
|
for (std::uint64_t i = 0; i < bits.size(); i++) {
|
|
bvb.set_bit(i, bits[i]);
|
|
}
|
|
for (std::uint64_t i = 0; i < bits.size(); i++) {
|
|
REQUIRE_EQ(bvb[i], bits[i]);
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Test bit_vector::builder with push_back") {
|
|
const auto bits = xcdat::test::make_random_bits(10000);
|
|
|
|
xcdat::bit_vector::builder bvb;
|
|
bvb.reserve(bits.size());
|
|
|
|
for (std::uint64_t i = 0; i < bits.size(); i++) {
|
|
bvb.push_back(bits[i]);
|
|
}
|
|
|
|
REQUIRE_EQ(bvb.size(), bits.size());
|
|
|
|
for (std::uint64_t i = 0; i < bits.size(); i++) {
|
|
REQUIRE_EQ(bvb[i], bits[i]);
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Test rank/select operations") {
|
|
const auto bits = xcdat::test::make_random_bits(10000);
|
|
test_rank_select(bits);
|
|
}
|
|
|
|
TEST_CASE("Test rank/select operations (all zeros)") {
|
|
const auto bits = xcdat::test::make_random_bits(10000, 0.0);
|
|
test_rank_select(bits);
|
|
}
|
|
|
|
TEST_CASE("Test rank/select operations (all ones)") {
|
|
const auto bits = xcdat::test::make_random_bits(10000, 1.1);
|
|
test_rank_select(bits);
|
|
}
|