forked from OSchip/llvm-project
90 lines
2.2 KiB
C++
90 lines
2.2 KiB
C++
//===-- sanitizer_lzw_test.cpp ----------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#include "sanitizer_common/sanitizer_lzw.h"
|
|
|
|
#include <iterator>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "sanitizer_hash.h"
|
|
|
|
namespace __sanitizer {
|
|
|
|
template <typename T>
|
|
struct LzwTest : public ::testing::Test {
|
|
template <typename Generator>
|
|
void Run(size_t n, Generator gen) {
|
|
std::vector<T> data(n);
|
|
std::generate(data.begin(), data.end(), gen);
|
|
|
|
std::vector<u64> lzw;
|
|
LzwEncode<T>(data.begin(), data.end(), std::back_inserter(lzw));
|
|
|
|
std::vector<T> unlzw(data.size() * 2);
|
|
auto unlzw_end = LzwDecode<T>(lzw.begin(), lzw.end(), unlzw.data());
|
|
unlzw.resize(unlzw_end - unlzw.data());
|
|
|
|
EXPECT_EQ(data, unlzw);
|
|
}
|
|
};
|
|
|
|
static constexpr size_t kSizes[] = {0, 1, 2, 7, 13, 32, 129, 10000};
|
|
|
|
using LzwTestTypes = ::testing::Types<u8, u16, u32, u64>;
|
|
TYPED_TEST_SUITE(LzwTest, LzwTestTypes, );
|
|
|
|
TYPED_TEST(LzwTest, Same) {
|
|
MurMur2Hash64Builder h(0);
|
|
for (size_t sz : kSizes) {
|
|
u64 v = 0;
|
|
for (size_t i = 0; i < 100 && !this->HasFailure(); ++i) {
|
|
this->Run(sz, [&] { return v; });
|
|
h.add(i);
|
|
v = h.get();
|
|
}
|
|
}
|
|
}
|
|
|
|
TYPED_TEST(LzwTest, Increment) {
|
|
MurMur2Hash64Builder h(0);
|
|
for (size_t sz : kSizes) {
|
|
u64 v = 0;
|
|
for (size_t i = 0; i < 100 && !this->HasFailure(); ++i) {
|
|
this->Run(sz, [&v] { return v++; });
|
|
h.add(i);
|
|
v = h.get();
|
|
}
|
|
}
|
|
}
|
|
|
|
TYPED_TEST(LzwTest, IncrementMod) {
|
|
MurMur2Hash64Builder h(0);
|
|
for (size_t sz : kSizes) {
|
|
u64 v = 0;
|
|
for (size_t i = 1; i < 16 && !this->HasFailure(); ++i) {
|
|
this->Run(sz, [&] { return v++ % i; });
|
|
h.add(i);
|
|
v = h.get();
|
|
}
|
|
}
|
|
}
|
|
|
|
TYPED_TEST(LzwTest, RandomLimited) {
|
|
for (size_t sz : kSizes) {
|
|
for (size_t i = 1; i < 1000 && !this->HasFailure(); i *= 2) {
|
|
u64 v = 0;
|
|
this->Run(sz, [&] {
|
|
MurMur2Hash64Builder h(v % i /* Keep unique set limited */);
|
|
v = h.get();
|
|
return v;
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace __sanitizer
|