Repo created

3party/succinct/test_gamma_bit_vector.cpp

@@ -0,0 +1,62 @@
+#define BOOST_TEST_MODULE gamma_bit_vector
+#include "test_common.hpp"
+
+#include <cstdlib>
+
+#include "gamma_bit_vector.hpp"
+
+typedef std::vector<uint64_t> std_vector_type;
+
+std_vector_type random_vector(size_t test_size)
+{
+    std_vector_type v;
+
+    for (size_t i = 0; i < test_size; ++i) {
+        bool b = uint64_t(rand()) & 1;
+        if (rand() < (RAND_MAX / 3)) {
+            v.push_back(b);
+        } else {
+            v.push_back((uint64_t(rand()) << 1) | b);
+        }
+    }
+
+    return v;
+}
+
+BOOST_AUTO_TEST_CASE(gamma_bit_vector)
+{
+    srand(42);
+    const size_t test_size = 12345;
+    std_vector_type v = random_vector(test_size);
+
+    succinct::gamma_bit_vector vv(v);
+
+    BOOST_REQUIRE_EQUAL(v.size(), vv.size());
+    for (size_t i = 0; i < v.size(); ++i) {
+        MY_REQUIRE_EQUAL(v[i], vv[i], "i = " << i);
+    }
+}
+
+BOOST_AUTO_TEST_CASE(gamma_bit_enumerator)
+{
+    srand(42);
+    const size_t test_size = 12345;
+    std_vector_type v = random_vector(test_size);
+
+    succinct::gamma_bit_vector vv(v);
+
+    size_t i = 0;
+    size_t pos = 0;
+
+    succinct::forward_enumerator<succinct::gamma_bit_vector> e(vv, pos);
+    while (pos < vv.size()) {
+        succinct::gamma_bit_vector::value_type next = e.next();
+        MY_REQUIRE_EQUAL(next, v[pos], "pos = " << pos << " i = " << i);
+        pos += 1;
+
+        size_t step = uint64_t(rand()) % (vv.size() - pos + 1);
+        pos += step;
+        e = succinct::forward_enumerator<succinct::gamma_bit_vector>(vv, pos);
+        i += 1;
+    }
+}