#include <catch2/catch.hpp>
#include <fstream>
#include <quantiles_sketch.hpp>
namespace datasketches {
TEST_CASE("quantiles sketch double generate", "[serialize_for_java]") {
const unsigned n_arr[] = {0, 1, 10, 100, 1000, 10000, 100000, 1000000};
for (const unsigned n: n_arr) {
quantiles_sketch<double> sketch;
for (unsigned i = 1; i <= n; ++i) sketch.update(i);
std::ofstream os("quantiles_double_n" + std::to_string(n) + "_cpp.sk", std::ios::binary);
sketch.serialize(os);
}
}
struct compare_as_number {
bool operator()(const std::string& a, const std::string& b) const {
return std::stoi(a) < std::stoi(b);
}
};
TEST_CASE("quantiles sketch string generate", "[serialize_for_java]") {
const unsigned n_arr[] = {0, 1, 10, 100, 1000, 10000, 100000, 1000000};
for (const unsigned n: n_arr) {
quantiles_sketch<std::string, compare_as_number> sketch;
for (unsigned i = 1; i <= n; ++i) sketch.update(std::to_string(i));
std::ofstream os("quantiles_string_n" + std::to_string(n) + "_cpp.sk", std::ios::binary);
sketch.serialize(os);
}
}
}