#include <catch2/catch.hpp>
#include <bit_packing.hpp>
namespace datasketches {
static const uint64_t IGOLDEN64 = 0x9e3779b97f4a7c13ULL;
TEST_CASE("pack unpack bits") {
uint64_t value = 0xaa55aa55aa55aa55ULL; for (int m = 0; m < 10000; ++m) {
for (uint8_t bits = 1; bits <= 63; ++bits) {
int n = 8;
const uint64_t mask = (1ULL << bits) - 1;
std::vector<uint64_t> input(n, 0);
for (int i = 0; i < n; ++i) {
input[i] = value & mask;
value += IGOLDEN64;
}
std::vector<uint8_t> bytes(n * sizeof(uint64_t), 0);
uint8_t offset = 0;
uint8_t* ptr = bytes.data();
for (int i = 0; i < n; ++i) {
offset = pack_bits(input[i], bits, ptr, offset);
}
std::vector<uint64_t> output(n, 0);
offset = 0;
const uint8_t* cptr = bytes.data();
for (int i = 0; i < n; ++i) {
offset = unpack_bits(output[i], bits, cptr, offset);
}
for (int i = 0; i < n; ++i) {
REQUIRE(input[i] == output[i]);
}
}
}
}
TEST_CASE("pack unpack blocks") {
uint64_t value = 0xaa55aa55aa55aa55ULL; for (int n = 0; n < 10000; ++n) {
for (uint8_t bits = 1; bits <= 63; ++bits) {
const uint64_t mask = (1ULL << bits) - 1;
std::vector<uint64_t> input(8, 0);
for (int i = 0; i < 8; ++i) {
input[i] = value & mask;
value += IGOLDEN64;
}
std::vector<uint8_t> bytes(bits, 0);
pack_bits_block8(input.data(), bytes.data(), bits);
std::vector<uint64_t> output(8, 0);
unpack_bits_block8(output.data(), bytes.data(), bits);
for (int i = 0; i < 8; ++i) {
REQUIRE(input[i] == output[i]);
}
}
}
}
}