update

tests/test_tail_vector.cpp

@@ -0,0 +1,51 @@
+#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(std::move(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);
+}