#include <catch2/catch.hpp>
#include <fstream>
#include "array_of_doubles_sketch.hpp"
namespace datasketches {
TEST_CASE("aod sketch generate one value", "[serialize_for_java]") {
const unsigned n_arr[] = {0, 1, 10, 100, 1000, 10000, 100000, 1000000};
for (const unsigned n: n_arr) {
auto sketch = update_array_of_doubles_sketch::builder().build();
for (unsigned i = 0; i < n; ++i) sketch.update(i, std::vector<double>(1, i));
REQUIRE(sketch.is_empty() == (n == 0));
REQUIRE(sketch.get_estimate() == Approx(n).margin(n * 0.03));
std::ofstream os("aod_1_n" + std::to_string(n) + "_cpp.sk", std::ios::binary);
sketch.compact().serialize(os);
}
}
TEST_CASE("aod sketch generate three values", "[serialize_for_java]") {
const unsigned n_arr[] = {0, 1, 10, 100, 1000, 10000, 100000, 1000000};
for (const unsigned n: n_arr) {
auto sketch = update_array_of_doubles_sketch::builder(3).build();
for (unsigned i = 0; i < n; ++i) sketch.update(i, std::vector<double>(3, i));
REQUIRE(sketch.is_empty() == (n == 0));
REQUIRE(sketch.get_estimate() == Approx(n).margin(n * 0.03));
std::ofstream os("aod_3_n" + std::to_string(n) + "_cpp.sk", std::ios::binary);
sketch.compact().serialize(os);
}
}
TEST_CASE("aod sketch generate non-empty no entries", "[serialize_for_java]") {
auto sketch = update_array_of_doubles_sketch::builder().set_p(0.01).build();
sketch.update(1, std::vector<double>({1}));
REQUIRE_FALSE(sketch.is_empty());
REQUIRE(sketch.get_num_retained() == 0);
std::ofstream os("aod_1_non_empty_no_entries_cpp.sk", std::ios::binary);
sketch.compact().serialize(os);
}
}