52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
#define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
|
|
|
|
#include <algorithm>
|
|
#include <random>
|
|
|
|
#include "doctest/doctest.h"
|
|
#include "test_common.hpp"
|
|
#include "xcdat/tail_vector.hpp"
|
|
|
|
void test_tail_vector(const std::vector<std::string>& sufs, bool bin_mode = false) {
|
|
xcdat::tail_vector tvec;
|
|
std::vector<std::uint64_t> idxs(sufs.size());
|
|
|
|
{
|
|
xcdat::tail_vector::builder tvb;
|
|
for (std::uint64_t i = 0; i < sufs.size(); i++) {
|
|
tvb.set_suffix(sufs[i], i);
|
|
}
|
|
tvb.complete(bin_mode, [&](std::uint64_t npos, std::uint64_t tpos) { idxs[npos] = tpos; });
|
|
tvec = xcdat::tail_vector(tvb);
|
|
}
|
|
|
|
for (std::uint64_t i = 0; i < sufs.size(); i++) {
|
|
REQUIRE(tvec.match(sufs[i], idxs[i]));
|
|
}
|
|
for (std::uint64_t i = 0; i < sufs.size(); i++) {
|
|
std::string decoded;
|
|
tvec.decode(idxs[i], [&](char c) { decoded.push_back(c); });
|
|
REQUIRE_EQ(sufs[i], decoded);
|
|
}
|
|
}
|
|
|
|
TEST_CASE("Test xcdat::tail_vector (tiny)") {
|
|
std::vector<std::string> sufs = {"ML", "STATS", "A", "M", "L", "AKDD", "M", "R", "DD", "OD"};
|
|
test_tail_vector(sufs);
|
|
}
|
|
|
|
TEST_CASE("Test xcdat::tail_vector (random, A--B)") {
|
|
std::vector<std::string> sufs = xcdat::test::make_random_keys(10000, 1, 30, 'A', 'B');
|
|
test_tail_vector(sufs);
|
|
}
|
|
|
|
TEST_CASE("Test xcdat::tail_vector (random, A--Z)") {
|
|
std::vector<std::string> sufs = xcdat::test::make_random_keys(10000, 1, 30, 'A', 'Z');
|
|
test_tail_vector(sufs);
|
|
}
|
|
|
|
TEST_CASE("Test xcdat::tail_vector (random, 0x00--0xFF)") {
|
|
std::vector<std::string> sufs = xcdat::test::make_random_keys(10000, 1, 30, INT8_MIN, INT8_MAX);
|
|
test_tail_vector(sufs, true);
|
|
}
|