datasketches 0.3.2 → 0.4.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/CHANGELOG.md +6 -0
- data/NOTICE +1 -1
- data/README.md +0 -2
- data/ext/datasketches/cpc_wrapper.cpp +2 -2
- data/ext/datasketches/kll_wrapper.cpp +0 -10
- data/lib/datasketches/version.rb +1 -1
- data/lib/datasketches.rb +1 -1
- data/vendor/datasketches-cpp/CMakeLists.txt +1 -0
- data/vendor/datasketches-cpp/CODE_OF_CONDUCT.md +3 -0
- data/vendor/datasketches-cpp/CONTRIBUTING.md +50 -0
- data/vendor/datasketches-cpp/Doxyfile +2827 -0
- data/vendor/datasketches-cpp/LICENSE +0 -76
- data/vendor/datasketches-cpp/NOTICE +1 -1
- data/vendor/datasketches-cpp/README.md +1 -3
- data/vendor/datasketches-cpp/common/CMakeLists.txt +12 -11
- data/vendor/datasketches-cpp/common/include/common_defs.hpp +11 -8
- data/vendor/datasketches-cpp/common/include/count_zeros.hpp +0 -2
- data/vendor/datasketches-cpp/common/include/kolmogorov_smirnov.hpp +9 -6
- data/vendor/datasketches-cpp/common/include/optional.hpp +148 -0
- data/vendor/datasketches-cpp/common/include/quantiles_sorted_view.hpp +95 -2
- data/vendor/datasketches-cpp/common/include/quantiles_sorted_view_impl.hpp +1 -1
- data/vendor/datasketches-cpp/common/include/serde.hpp +69 -20
- data/vendor/datasketches-cpp/common/test/CMakeLists.txt +1 -1
- data/vendor/datasketches-cpp/common/test/optional_test.cpp +85 -0
- data/vendor/datasketches-cpp/common/test/test_allocator.hpp +14 -14
- data/vendor/datasketches-cpp/count/include/count_min.hpp +132 -78
- data/vendor/datasketches-cpp/count/include/count_min_impl.hpp +132 -152
- data/vendor/datasketches-cpp/count/test/CMakeLists.txt +11 -12
- data/vendor/datasketches-cpp/count/test/count_min_allocation_test.cpp +61 -61
- data/vendor/datasketches-cpp/count/test/count_min_test.cpp +175 -178
- data/vendor/datasketches-cpp/cpc/include/cpc_common.hpp +14 -20
- data/vendor/datasketches-cpp/cpc/include/cpc_compressor.hpp +7 -4
- data/vendor/datasketches-cpp/cpc/include/cpc_compressor_impl.hpp +17 -17
- data/vendor/datasketches-cpp/cpc/include/cpc_sketch.hpp +40 -40
- data/vendor/datasketches-cpp/cpc/include/cpc_sketch_impl.hpp +13 -10
- data/vendor/datasketches-cpp/cpc/include/cpc_union.hpp +35 -11
- data/vendor/datasketches-cpp/cpc/include/cpc_union_impl.hpp +8 -8
- data/vendor/datasketches-cpp/cpc/include/u32_table.hpp +3 -2
- data/vendor/datasketches-cpp/cpc/include/u32_table_impl.hpp +5 -5
- data/vendor/datasketches-cpp/cpc/test/CMakeLists.txt +20 -7
- data/vendor/datasketches-cpp/cpc/test/cpc_sketch_deserialize_from_java_test.cpp +60 -0
- data/vendor/datasketches-cpp/{python/include/py_object_lt.hpp → cpc/test/cpc_sketch_serialize_for_java.cpp} +15 -14
- data/vendor/datasketches-cpp/cpc/test/cpc_sketch_test.cpp +4 -29
- data/vendor/datasketches-cpp/cpc/test/cpc_union_test.cpp +4 -4
- data/vendor/datasketches-cpp/density/include/density_sketch.hpp +29 -9
- data/vendor/datasketches-cpp/density/include/density_sketch_impl.hpp +1 -1
- data/vendor/datasketches-cpp/density/test/CMakeLists.txt +0 -1
- data/vendor/datasketches-cpp/fi/include/frequent_items_sketch.hpp +21 -9
- data/vendor/datasketches-cpp/fi/include/frequent_items_sketch_impl.hpp +6 -4
- data/vendor/datasketches-cpp/fi/test/CMakeLists.txt +14 -1
- data/vendor/datasketches-cpp/fi/test/frequent_items_sketch_deserialize_from_java_test.cpp +95 -0
- data/vendor/datasketches-cpp/fi/test/frequent_items_sketch_serialize_for_java.cpp +83 -0
- data/vendor/datasketches-cpp/fi/test/frequent_items_sketch_test.cpp +3 -42
- data/vendor/datasketches-cpp/hll/include/CouponList-internal.hpp +2 -2
- data/vendor/datasketches-cpp/hll/include/CouponList.hpp +3 -1
- data/vendor/datasketches-cpp/hll/include/HllArray-internal.hpp +3 -3
- data/vendor/datasketches-cpp/hll/include/HllArray.hpp +5 -3
- data/vendor/datasketches-cpp/hll/include/HllSketch-internal.hpp +4 -4
- data/vendor/datasketches-cpp/hll/include/HllSketchImpl.hpp +3 -1
- data/vendor/datasketches-cpp/hll/include/HllUtil.hpp +0 -12
- data/vendor/datasketches-cpp/hll/include/hll.hpp +70 -57
- data/vendor/datasketches-cpp/hll/test/CMakeLists.txt +14 -1
- data/vendor/datasketches-cpp/hll/test/ToFromByteArrayTest.cpp +0 -68
- data/vendor/datasketches-cpp/hll/test/hll_sketch_deserialize_from_java_test.cpp +69 -0
- data/vendor/datasketches-cpp/hll/test/hll_sketch_serialize_for_java.cpp +52 -0
- data/vendor/datasketches-cpp/kll/include/kll_helper_impl.hpp +2 -2
- data/vendor/datasketches-cpp/kll/include/kll_sketch.hpp +71 -50
- data/vendor/datasketches-cpp/kll/include/kll_sketch_impl.hpp +59 -130
- data/vendor/datasketches-cpp/kll/test/CMakeLists.txt +14 -1
- data/vendor/datasketches-cpp/kll/test/kll_sketch_deserialize_from_java_test.cpp +103 -0
- data/vendor/datasketches-cpp/kll/test/kll_sketch_serialize_for_java.cpp +62 -0
- data/vendor/datasketches-cpp/kll/test/kll_sketch_test.cpp +3 -38
- data/vendor/datasketches-cpp/quantiles/include/quantiles_sketch.hpp +68 -51
- data/vendor/datasketches-cpp/quantiles/include/quantiles_sketch_impl.hpp +62 -132
- data/vendor/datasketches-cpp/quantiles/test/CMakeLists.txt +14 -1
- data/vendor/datasketches-cpp/quantiles/test/quantiles_sketch_deserialize_from_java_test.cpp +84 -0
- data/vendor/datasketches-cpp/quantiles/test/quantiles_sketch_serialize_for_java.cpp +52 -0
- data/vendor/datasketches-cpp/quantiles/test/quantiles_sketch_test.cpp +14 -38
- data/vendor/datasketches-cpp/req/include/req_common.hpp +7 -3
- data/vendor/datasketches-cpp/req/include/req_compactor_impl.hpp +2 -2
- data/vendor/datasketches-cpp/req/include/req_sketch.hpp +97 -23
- data/vendor/datasketches-cpp/req/include/req_sketch_impl.hpp +48 -109
- data/vendor/datasketches-cpp/req/test/CMakeLists.txt +14 -1
- data/vendor/datasketches-cpp/req/test/req_sketch_deserialize_from_java_test.cpp +55 -0
- data/vendor/datasketches-cpp/{tuple/include/array_of_doubles_intersection_impl.hpp → req/test/req_sketch_serialize_for_java.cpp} +12 -7
- data/vendor/datasketches-cpp/req/test/req_sketch_test.cpp +3 -89
- data/vendor/datasketches-cpp/sampling/CMakeLists.txt +4 -0
- data/vendor/datasketches-cpp/sampling/include/ebpps_sample.hpp +210 -0
- data/vendor/datasketches-cpp/sampling/include/ebpps_sample_impl.hpp +535 -0
- data/vendor/datasketches-cpp/sampling/include/ebpps_sketch.hpp +281 -0
- data/vendor/datasketches-cpp/sampling/include/ebpps_sketch_impl.hpp +531 -0
- data/vendor/datasketches-cpp/sampling/include/var_opt_sketch.hpp +69 -26
- data/vendor/datasketches-cpp/sampling/include/var_opt_sketch_impl.hpp +3 -3
- data/vendor/datasketches-cpp/sampling/include/var_opt_union.hpp +10 -11
- data/vendor/datasketches-cpp/sampling/include/var_opt_union_impl.hpp +4 -4
- data/vendor/datasketches-cpp/sampling/test/CMakeLists.txt +55 -8
- data/vendor/datasketches-cpp/sampling/test/ebpps_allocation_test.cpp +96 -0
- data/vendor/datasketches-cpp/sampling/test/ebpps_sample_test.cpp +137 -0
- data/vendor/datasketches-cpp/sampling/test/ebpps_sketch_test.cpp +266 -0
- data/vendor/datasketches-cpp/sampling/test/var_opt_sketch_deserialize_from_java_test.cpp +81 -0
- data/vendor/datasketches-cpp/sampling/test/var_opt_sketch_serialize_for_java.cpp +54 -0
- data/vendor/datasketches-cpp/sampling/test/var_opt_sketch_test.cpp +0 -37
- data/vendor/datasketches-cpp/sampling/test/var_opt_union_deserialize_from_java_test.cpp +50 -0
- data/vendor/datasketches-cpp/sampling/test/var_opt_union_serialize_for_java.cpp +56 -0
- data/vendor/datasketches-cpp/sampling/test/var_opt_union_test.cpp +0 -18
- data/vendor/datasketches-cpp/theta/include/bit_packing.hpp +2608 -2608
- data/vendor/datasketches-cpp/theta/include/bounds_on_ratios_in_sampled_sets.hpp +1 -0
- data/vendor/datasketches-cpp/theta/include/bounds_on_ratios_in_theta_sketched_sets.hpp +7 -6
- data/vendor/datasketches-cpp/theta/include/theta_a_not_b.hpp +20 -5
- data/vendor/datasketches-cpp/theta/include/theta_constants.hpp +10 -4
- data/vendor/datasketches-cpp/theta/include/theta_helpers.hpp +1 -1
- data/vendor/datasketches-cpp/theta/include/theta_intersection.hpp +13 -5
- data/vendor/datasketches-cpp/theta/include/theta_intersection_base_impl.hpp +5 -5
- data/vendor/datasketches-cpp/theta/include/theta_intersection_impl.hpp +3 -3
- data/vendor/datasketches-cpp/theta/include/theta_jaccard_similarity.hpp +2 -1
- data/vendor/datasketches-cpp/theta/include/theta_jaccard_similarity_base.hpp +1 -0
- data/vendor/datasketches-cpp/theta/include/theta_set_difference_base_impl.hpp +1 -1
- data/vendor/datasketches-cpp/theta/include/theta_sketch.hpp +126 -27
- data/vendor/datasketches-cpp/theta/include/theta_sketch_impl.hpp +8 -8
- data/vendor/datasketches-cpp/theta/include/theta_union.hpp +17 -10
- data/vendor/datasketches-cpp/theta/include/theta_union_base_impl.hpp +1 -1
- data/vendor/datasketches-cpp/theta/include/theta_union_impl.hpp +3 -3
- data/vendor/datasketches-cpp/theta/include/theta_update_sketch_base.hpp +5 -2
- data/vendor/datasketches-cpp/theta/include/theta_update_sketch_base_impl.hpp +11 -1
- data/vendor/datasketches-cpp/theta/test/CMakeLists.txt +14 -1
- data/vendor/datasketches-cpp/theta/test/theta_sketch_deserialize_from_java_test.cpp +57 -0
- data/vendor/datasketches-cpp/theta/test/theta_sketch_serialize_for_java.cpp +61 -0
- data/vendor/datasketches-cpp/theta/test/theta_sketch_test.cpp +0 -188
- data/vendor/datasketches-cpp/tuple/CMakeLists.txt +8 -7
- data/vendor/datasketches-cpp/tuple/include/array_of_doubles_sketch.hpp +19 -144
- data/vendor/datasketches-cpp/tuple/include/{array_of_doubles_a_not_b.hpp → array_tuple_a_not_b.hpp} +24 -16
- data/vendor/datasketches-cpp/tuple/include/{array_of_doubles_a_not_b_impl.hpp → array_tuple_a_not_b_impl.hpp} +4 -4
- data/vendor/datasketches-cpp/tuple/include/array_tuple_intersection.hpp +65 -0
- data/vendor/datasketches-cpp/{python/include/py_object_ostream.hpp → tuple/include/array_tuple_intersection_impl.hpp} +7 -24
- data/vendor/datasketches-cpp/tuple/include/array_tuple_sketch.hpp +237 -0
- data/vendor/datasketches-cpp/tuple/include/{array_of_doubles_sketch_impl.hpp → array_tuple_sketch_impl.hpp} +40 -41
- data/vendor/datasketches-cpp/tuple/include/array_tuple_union.hpp +81 -0
- data/vendor/datasketches-cpp/tuple/include/array_tuple_union_impl.hpp +43 -0
- data/vendor/datasketches-cpp/tuple/include/tuple_a_not_b.hpp +11 -2
- data/vendor/datasketches-cpp/tuple/include/tuple_intersection.hpp +17 -10
- data/vendor/datasketches-cpp/tuple/include/tuple_jaccard_similarity.hpp +2 -1
- data/vendor/datasketches-cpp/tuple/include/tuple_sketch.hpp +95 -32
- data/vendor/datasketches-cpp/tuple/include/tuple_union.hpp +19 -11
- data/vendor/datasketches-cpp/tuple/test/CMakeLists.txt +16 -1
- data/vendor/datasketches-cpp/tuple/test/aod_sketch_deserialize_from_java_test.cpp +76 -0
- data/vendor/datasketches-cpp/tuple/test/aod_sketch_serialize_for_java.cpp +62 -0
- data/vendor/datasketches-cpp/tuple/test/array_of_doubles_sketch_test.cpp +5 -129
- data/vendor/datasketches-cpp/tuple/test/engagement_test.cpp +85 -89
- data/vendor/datasketches-cpp/tuple/test/tuple_jaccard_similarity_test.cpp +3 -1
- data/vendor/datasketches-cpp/tuple/test/tuple_sketch_deserialize_from_java_test.cpp +47 -0
- data/vendor/datasketches-cpp/tuple/test/tuple_sketch_serialize_for_java.cpp +38 -0
- data/vendor/datasketches-cpp/tuple/test/tuple_sketch_test.cpp +1 -1
- data/vendor/datasketches-cpp/version.cfg.in +1 -1
- metadata +47 -93
- data/vendor/datasketches-cpp/MANIFEST.in +0 -39
- data/vendor/datasketches-cpp/fi/test/items_sketch_string_from_java.sk +0 -0
- data/vendor/datasketches-cpp/fi/test/items_sketch_string_utf8_from_java.sk +0 -0
- data/vendor/datasketches-cpp/fi/test/longs_sketch_from_java.sk +0 -0
- data/vendor/datasketches-cpp/hll/test/array6_from_java.sk +0 -0
- data/vendor/datasketches-cpp/hll/test/compact_array4_from_java.sk +0 -0
- data/vendor/datasketches-cpp/hll/test/compact_set_from_java.sk +0 -0
- data/vendor/datasketches-cpp/hll/test/list_from_java.sk +0 -0
- data/vendor/datasketches-cpp/hll/test/updatable_array4_from_java.sk +0 -0
- data/vendor/datasketches-cpp/hll/test/updatable_set_from_java.sk +0 -0
- data/vendor/datasketches-cpp/kll/test/kll_sketch_from_java.sk +0 -0
- data/vendor/datasketches-cpp/pyproject.toml +0 -23
- data/vendor/datasketches-cpp/python/CMakeLists.txt +0 -87
- data/vendor/datasketches-cpp/python/README.md +0 -85
- data/vendor/datasketches-cpp/python/datasketches/DensityWrapper.py +0 -87
- data/vendor/datasketches-cpp/python/datasketches/KernelFunction.py +0 -35
- data/vendor/datasketches-cpp/python/datasketches/PySerDe.py +0 -110
- data/vendor/datasketches-cpp/python/datasketches/TuplePolicy.py +0 -77
- data/vendor/datasketches-cpp/python/datasketches/TupleWrapper.py +0 -205
- data/vendor/datasketches-cpp/python/datasketches/__init__.py +0 -38
- data/vendor/datasketches-cpp/python/include/kernel_function.hpp +0 -98
- data/vendor/datasketches-cpp/python/include/py_serde.hpp +0 -113
- data/vendor/datasketches-cpp/python/include/quantile_conditional.hpp +0 -104
- data/vendor/datasketches-cpp/python/include/tuple_policy.hpp +0 -136
- data/vendor/datasketches-cpp/python/jupyter/CPCSketch.ipynb +0 -345
- data/vendor/datasketches-cpp/python/jupyter/FrequentItemsSketch.ipynb +0 -354
- data/vendor/datasketches-cpp/python/jupyter/HLLSketch.ipynb +0 -346
- data/vendor/datasketches-cpp/python/jupyter/KLLSketch.ipynb +0 -463
- data/vendor/datasketches-cpp/python/jupyter/ThetaSketchNotebook.ipynb +0 -403
- data/vendor/datasketches-cpp/python/pybind11Path.cmd +0 -21
- data/vendor/datasketches-cpp/python/src/__init__.py +0 -18
- data/vendor/datasketches-cpp/python/src/count_wrapper.cpp +0 -101
- data/vendor/datasketches-cpp/python/src/cpc_wrapper.cpp +0 -76
- data/vendor/datasketches-cpp/python/src/datasketches.cpp +0 -58
- data/vendor/datasketches-cpp/python/src/density_wrapper.cpp +0 -95
- data/vendor/datasketches-cpp/python/src/fi_wrapper.cpp +0 -182
- data/vendor/datasketches-cpp/python/src/hll_wrapper.cpp +0 -126
- data/vendor/datasketches-cpp/python/src/kll_wrapper.cpp +0 -158
- data/vendor/datasketches-cpp/python/src/ks_wrapper.cpp +0 -68
- data/vendor/datasketches-cpp/python/src/py_serde.cpp +0 -112
- data/vendor/datasketches-cpp/python/src/quantiles_wrapper.cpp +0 -155
- data/vendor/datasketches-cpp/python/src/req_wrapper.cpp +0 -154
- data/vendor/datasketches-cpp/python/src/theta_wrapper.cpp +0 -166
- data/vendor/datasketches-cpp/python/src/tuple_wrapper.cpp +0 -215
- data/vendor/datasketches-cpp/python/src/vector_of_kll.cpp +0 -490
- data/vendor/datasketches-cpp/python/src/vo_wrapper.cpp +0 -173
- data/vendor/datasketches-cpp/python/tests/__init__.py +0 -16
- data/vendor/datasketches-cpp/python/tests/count_min_test.py +0 -86
- data/vendor/datasketches-cpp/python/tests/cpc_test.py +0 -64
- data/vendor/datasketches-cpp/python/tests/density_test.py +0 -93
- data/vendor/datasketches-cpp/python/tests/fi_test.py +0 -149
- data/vendor/datasketches-cpp/python/tests/hll_test.py +0 -129
- data/vendor/datasketches-cpp/python/tests/kll_test.py +0 -159
- data/vendor/datasketches-cpp/python/tests/quantiles_test.py +0 -160
- data/vendor/datasketches-cpp/python/tests/req_test.py +0 -159
- data/vendor/datasketches-cpp/python/tests/theta_test.py +0 -148
- data/vendor/datasketches-cpp/python/tests/tuple_test.py +0 -206
- data/vendor/datasketches-cpp/python/tests/vector_of_kll_test.py +0 -148
- data/vendor/datasketches-cpp/python/tests/vo_test.py +0 -132
- data/vendor/datasketches-cpp/req/test/req_float_empty_from_java.sk +0 -0
- data/vendor/datasketches-cpp/req/test/req_float_estimation_from_java.sk +0 -0
- data/vendor/datasketches-cpp/req/test/req_float_exact_from_java.sk +0 -0
- data/vendor/datasketches-cpp/req/test/req_float_raw_items_from_java.sk +0 -0
- data/vendor/datasketches-cpp/req/test/req_float_single_item_from_java.sk +0 -0
- data/vendor/datasketches-cpp/sampling/test/binaries_from_java.txt +0 -67
- data/vendor/datasketches-cpp/sampling/test/varopt_sketch_long_sampling.sk +0 -0
- data/vendor/datasketches-cpp/sampling/test/varopt_sketch_string_exact.sk +0 -0
- data/vendor/datasketches-cpp/sampling/test/varopt_union_double_sampling.sk +0 -0
- data/vendor/datasketches-cpp/setup.py +0 -110
- data/vendor/datasketches-cpp/theta/test/theta_compact_empty_from_java.sk +0 -0
- data/vendor/datasketches-cpp/theta/test/theta_compact_estimation_from_java.sk +0 -0
- data/vendor/datasketches-cpp/theta/test/theta_compact_exact_from_java.sk +0 -0
- data/vendor/datasketches-cpp/theta/test/theta_compact_single_item_from_java.sk +0 -0
- data/vendor/datasketches-cpp/tox.ini +0 -26
- data/vendor/datasketches-cpp/tuple/include/array_of_doubles_intersection.hpp +0 -52
- data/vendor/datasketches-cpp/tuple/include/array_of_doubles_union.hpp +0 -81
- data/vendor/datasketches-cpp/tuple/include/array_of_doubles_union_impl.hpp +0 -43
- data/vendor/datasketches-cpp/tuple/test/aod_1_compact_empty_from_java.sk +0 -1
- data/vendor/datasketches-cpp/tuple/test/aod_1_compact_estimation_from_java.sk +0 -0
- data/vendor/datasketches-cpp/tuple/test/aod_1_compact_non_empty_no_entries_from_java.sk +0 -0
- data/vendor/datasketches-cpp/tuple/test/aod_2_compact_exact_from_java.sk +0 -0
- data/vendor/datasketches-cpp/tuple/test/aod_3_compact_empty_from_java.sk +0 -1
@@ -1,86 +0,0 @@
|
|
1
|
-
# Licensed to the Apache Software Foundation (ASF) under one
|
2
|
-
# or more contributor license agreements. See the NOTICE file
|
3
|
-
# distributed with this work for additional information
|
4
|
-
# regarding copyright ownership. The ASF licenses this file
|
5
|
-
# to you under the Apache License, Version 2.0 (the
|
6
|
-
# "License"); you may not use this file except in compliance
|
7
|
-
# with the License. You may obtain a copy of the License at
|
8
|
-
#
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
10
|
-
#
|
11
|
-
# Unless required by applicable law or agreed to in writing,
|
12
|
-
# software distributed under the License is distributed on an
|
13
|
-
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
14
|
-
# KIND, either express or implied. See the License for the
|
15
|
-
# specific language governing permissions and limitations
|
16
|
-
# under the License.
|
17
|
-
|
18
|
-
import unittest
|
19
|
-
from datasketches import count_min_sketch
|
20
|
-
|
21
|
-
class CountMinTest(unittest.TestCase):
|
22
|
-
def test_count_min_example(self):
|
23
|
-
# we'll define target confidence and relative error and use the built-in
|
24
|
-
# methods to determine how many hashes and buckets to use
|
25
|
-
confidence = 0.95
|
26
|
-
num_hashes = count_min_sketch.suggest_num_hashes(confidence)
|
27
|
-
relative_error = 0.01
|
28
|
-
num_buckets = count_min_sketch.suggest_num_buckets(relative_error)
|
29
|
-
|
30
|
-
# now we can create a few empty sketches
|
31
|
-
cm = count_min_sketch(num_hashes, num_buckets)
|
32
|
-
cm2 = count_min_sketch(num_hashes, num_buckets)
|
33
|
-
self.assertTrue(cm.is_empty())
|
34
|
-
|
35
|
-
# we'll use a moderate number of distinct items with
|
36
|
-
# increasing weights, with each item's weight being
|
37
|
-
# equal to its value
|
38
|
-
n = 1000
|
39
|
-
total_wt = 0
|
40
|
-
for i in range(1, n+1):
|
41
|
-
cm.update(i, i)
|
42
|
-
total_wt += i
|
43
|
-
self.assertFalse(cm.is_empty())
|
44
|
-
self.assertEqual(cm.get_total_weight(), total_wt)
|
45
|
-
|
46
|
-
# querying the items, each of them should
|
47
|
-
# have a non-zero count. the estimate should
|
48
|
-
# be at least i with appropriately behaved bounds.
|
49
|
-
for i in range(1, n+1):
|
50
|
-
val = cm.get_estimate(i)
|
51
|
-
self.assertGreaterEqual(val, i)
|
52
|
-
self.assertGreaterEqual(val, cm.get_lower_bound(i))
|
53
|
-
self.assertGreater(cm.get_upper_bound(i), val)
|
54
|
-
|
55
|
-
# values not in the sketch should have lower estimates, but
|
56
|
-
# are not guaranteed to be zero and will succeed
|
57
|
-
self.assertIsNotNone(cm.get_estimate("not in set"))
|
58
|
-
|
59
|
-
# we can create another sketch with partial overlap
|
60
|
-
# and merge them
|
61
|
-
for i in range(int(n / 2), int(3 * n / 2)):
|
62
|
-
cm2.update(i, i)
|
63
|
-
cm.merge(cm2)
|
64
|
-
|
65
|
-
# and the estimated weight for the overlapped meerged values
|
66
|
-
# (n/2 to n) should now be at least 2x the value
|
67
|
-
self.assertGreaterEqual(cm.get_estimate(n), 2 * n)
|
68
|
-
|
69
|
-
# finally, serialize and reconstruct
|
70
|
-
cm_bytes = cm.serialize()
|
71
|
-
self.assertEqual(cm.get_serialized_size_bytes(), len(cm_bytes))
|
72
|
-
new_cm = count_min_sketch.deserialize(cm_bytes)
|
73
|
-
|
74
|
-
# and now interrogate the sketch
|
75
|
-
self.assertFalse(new_cm.is_empty())
|
76
|
-
self.assertEqual(new_cm.get_num_hashes(), cm.get_num_hashes())
|
77
|
-
self.assertEqual(new_cm.get_num_buckets(), cm.get_num_buckets())
|
78
|
-
self.assertEqual(new_cm.get_total_weight(), cm.get_total_weight())
|
79
|
-
|
80
|
-
# we can also iterate through values in and out of the sketch to ensure
|
81
|
-
# the estimates match
|
82
|
-
for i in range(0, 2 * n):
|
83
|
-
self.assertEqual(cm.get_estimate(i), new_cm.get_estimate(i))
|
84
|
-
|
85
|
-
if __name__ == '__main__':
|
86
|
-
unittest.main()
|
@@ -1,64 +0,0 @@
|
|
1
|
-
# Licensed to the Apache Software Foundation (ASF) under one
|
2
|
-
# or more contributor license agreements. See the NOTICE file
|
3
|
-
# distributed with this work for additional information
|
4
|
-
# regarding copyright ownership. The ASF licenses this file
|
5
|
-
# to you under the Apache License, Version 2.0 (the
|
6
|
-
# "License"); you may not use this file except in compliance
|
7
|
-
# with the License. You may obtain a copy of the License at
|
8
|
-
#
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
10
|
-
#
|
11
|
-
# Unless required by applicable law or agreed to in writing,
|
12
|
-
# software distributed under the License is distributed on an
|
13
|
-
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
14
|
-
# KIND, either express or implied. See the License for the
|
15
|
-
# specific language governing permissions and limitations
|
16
|
-
# under the License.
|
17
|
-
|
18
|
-
import unittest
|
19
|
-
from datasketches import cpc_sketch, cpc_union
|
20
|
-
|
21
|
-
class CpcTest(unittest.TestCase):
|
22
|
-
def test_cpc_example(self):
|
23
|
-
lgk = 12 # 2^k = 4096 rows in the table
|
24
|
-
n = 1 << 18 # ~256k distinct values
|
25
|
-
|
26
|
-
# create a couple sketches and inject some values
|
27
|
-
# we'll have 1/4 of the values overlap
|
28
|
-
cpc = cpc_sketch(lgk)
|
29
|
-
cpc2 = cpc_sketch(lgk)
|
30
|
-
offset = int(3 * n / 4) # it's a float w/o cast
|
31
|
-
# because we hash on the bits, not an abstract numeric value,
|
32
|
-
# cpc.update(1) and cpc.update(1.0) give different results.
|
33
|
-
for i in range(0, n):
|
34
|
-
cpc.update(i)
|
35
|
-
cpc2.update(i + offset)
|
36
|
-
|
37
|
-
# although we provide get_composite_estimate() and get_estimate(),
|
38
|
-
# the latter will always give the best available estimate. we
|
39
|
-
# recommend using get_estimate().
|
40
|
-
# we can check that the upper and lower bounds bracket the
|
41
|
-
# estimate, without needing to know the exact value.
|
42
|
-
self.assertLessEqual(cpc.get_lower_bound(1), cpc.get_estimate())
|
43
|
-
self.assertGreaterEqual(cpc.get_upper_bound(1), cpc.get_estimate())
|
44
|
-
|
45
|
-
# union is a separate class, so we need to get_result()
|
46
|
-
# to query the unioned sketches
|
47
|
-
union = cpc_union(lgk)
|
48
|
-
union.update(cpc)
|
49
|
-
union.update(cpc2)
|
50
|
-
result = union.get_result()
|
51
|
-
|
52
|
-
# since our process here (including post-union CPC) is
|
53
|
-
# deterministic, we have checked and know the exact
|
54
|
-
# answer is within one standard deviation of the estimate
|
55
|
-
self.assertLessEqual(result.get_lower_bound(1), 7 * n / 4)
|
56
|
-
self.assertGreaterEqual(result.get_upper_bound(1), 7 * n / 4)
|
57
|
-
|
58
|
-
# serialize for storage and reconstruct
|
59
|
-
sk_bytes = result.serialize()
|
60
|
-
new_cpc = cpc_sketch.deserialize(sk_bytes)
|
61
|
-
self.assertFalse(new_cpc.is_empty())
|
62
|
-
|
63
|
-
if __name__ == '__main__':
|
64
|
-
unittest.main()
|
@@ -1,93 +0,0 @@
|
|
1
|
-
# Licensed to the Apache Software Foundation (ASF) under one
|
2
|
-
# or more contributor license agreements. See the NOTICE file
|
3
|
-
# distributed with this work for additional information
|
4
|
-
# regarding copyright ownership. The ASF licenses this file
|
5
|
-
# to you under the Apache License, Version 2.0 (the
|
6
|
-
# "License"); you may not use this file except in compliance
|
7
|
-
# with the License. You may obtain a copy of the License at
|
8
|
-
#
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
10
|
-
#
|
11
|
-
# Unless required by applicable law or agreed to in writing,
|
12
|
-
# software distributed under the License is distributed on an
|
13
|
-
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
14
|
-
# KIND, either express or implied. See the License for the
|
15
|
-
# specific language governing permissions and limitations
|
16
|
-
# under the License.
|
17
|
-
|
18
|
-
import unittest
|
19
|
-
from datasketches import density_sketch, KernelFunction
|
20
|
-
import numpy as np
|
21
|
-
|
22
|
-
class UnitSphereKernel(KernelFunction):
|
23
|
-
def __call__(self, a: np.array, b: np.array) -> float:
|
24
|
-
if np.linalg.norm(a - b) < 1.0:
|
25
|
-
return 1.0
|
26
|
-
else:
|
27
|
-
return 0.0
|
28
|
-
|
29
|
-
class densityTest(unittest.TestCase):
|
30
|
-
def test_density_sketch(self):
|
31
|
-
k = 10
|
32
|
-
dim = 3
|
33
|
-
n = 1000
|
34
|
-
|
35
|
-
sketch = density_sketch(k, dim)
|
36
|
-
|
37
|
-
self.assertEqual(sketch.get_k(), k)
|
38
|
-
self.assertEqual(sketch.get_dim(), dim)
|
39
|
-
self.assertTrue(sketch.is_empty())
|
40
|
-
self.assertFalse(sketch.is_estimation_mode())
|
41
|
-
self.assertEqual(sketch.get_n(), 0)
|
42
|
-
self.assertEqual(sketch.get_num_retained(), 0)
|
43
|
-
|
44
|
-
for i in range(n):
|
45
|
-
sketch.update([i, i, i])
|
46
|
-
|
47
|
-
self.assertFalse(sketch.is_empty())
|
48
|
-
self.assertTrue(sketch.is_estimation_mode())
|
49
|
-
self.assertEqual(sketch.get_n(), n)
|
50
|
-
self.assertGreater(sketch.get_num_retained(), k)
|
51
|
-
self.assertLess(sketch.get_num_retained(), n)
|
52
|
-
self.assertGreater(sketch.get_estimate([n - 1, n - 1, n - 1]), 0)
|
53
|
-
|
54
|
-
for tuple in sketch:
|
55
|
-
vector = tuple[0]
|
56
|
-
weight = tuple[1]
|
57
|
-
self.assertEqual(len(vector), dim)
|
58
|
-
self.assertGreaterEqual(weight, 1)
|
59
|
-
|
60
|
-
sk_bytes = sketch.serialize()
|
61
|
-
sketch2 = density_sketch.deserialize(sk_bytes)
|
62
|
-
self.assertEqual(sketch.get_estimate([1.5, 2.5, 3.5]), sketch2.get_estimate([1.5, 2.5, 3.5]))
|
63
|
-
|
64
|
-
def test_density_merge(self):
|
65
|
-
sketch1 = density_sketch(10, 2)
|
66
|
-
sketch1.update([0, 0])
|
67
|
-
sketch2 = density_sketch(10, 2)
|
68
|
-
sketch2.update([0, 1])
|
69
|
-
sketch1.merge(sketch2)
|
70
|
-
self.assertEqual(sketch1.get_n(), 2)
|
71
|
-
self.assertEqual(sketch1.get_num_retained(), 2)
|
72
|
-
|
73
|
-
def test_custom_kernel(self):
|
74
|
-
gaussianSketch = density_sketch(10, 2) # default kernel
|
75
|
-
sphericalSketch = density_sketch(10, 2, UnitSphereKernel())
|
76
|
-
|
77
|
-
p = [1, 1]
|
78
|
-
gaussianSketch.update(p)
|
79
|
-
sphericalSketch.update(p)
|
80
|
-
|
81
|
-
# Spherical kernel should return 1.0 for a nearby point, 0 farther
|
82
|
-
# Gaussian kernel should return something nonzero when farther away
|
83
|
-
self.assertEqual(sphericalSketch.get_estimate([1.001, 1]), 1.0)
|
84
|
-
self.assertEqual(sphericalSketch.get_estimate([2, 2]), 0.0)
|
85
|
-
self.assertGreater(gaussianSketch.get_estimate([2, 2]), 0.0)
|
86
|
-
|
87
|
-
# We can also use a custom kernel when deserializing
|
88
|
-
sk_bytes = sphericalSketch.serialize()
|
89
|
-
sphericalRebuilt = density_sketch.deserialize(sk_bytes, UnitSphereKernel())
|
90
|
-
self.assertEqual(sphericalSketch.get_estimate([1.001, 1]), sphericalRebuilt.get_estimate([1.001, 1]))
|
91
|
-
|
92
|
-
if __name__ == '__main__':
|
93
|
-
unittest.main()
|
@@ -1,149 +0,0 @@
|
|
1
|
-
# Licensed to the Apache Software Foundation (ASF) under one
|
2
|
-
# or more contributor license agreements. See the NOTICE file
|
3
|
-
# distributed with this work for additional information
|
4
|
-
# regarding copyright ownership. The ASF licenses this file
|
5
|
-
# to you under the Apache License, Version 2.0 (the
|
6
|
-
# "License"); you may not use this file except in compliance
|
7
|
-
# with the License. You may obtain a copy of the License at
|
8
|
-
#
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
10
|
-
#
|
11
|
-
# Unless required by applicable law or agreed to in writing,
|
12
|
-
# software distributed under the License is distributed on an
|
13
|
-
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
14
|
-
# KIND, either express or implied. See the License for the
|
15
|
-
# specific language governing permissions and limitations
|
16
|
-
# under the License.
|
17
|
-
|
18
|
-
import unittest
|
19
|
-
from datasketches import frequent_strings_sketch, frequent_items_sketch
|
20
|
-
from datasketches import frequent_items_error_type, PyIntsSerDe
|
21
|
-
|
22
|
-
class FiTest(unittest.TestCase):
|
23
|
-
def test_fi_strings_example(self):
|
24
|
-
k = 3 # a small value so we can easily fill the sketch
|
25
|
-
fi = frequent_strings_sketch(k)
|
26
|
-
|
27
|
-
# we'll use a small number of distinct items so we
|
28
|
-
# can use exponentially increasing weights and have
|
29
|
-
# some frequent items, decreasing so we have some
|
30
|
-
# small items inserted after a purge
|
31
|
-
n = 8
|
32
|
-
for i in range(0, n):
|
33
|
-
fi.update(str(i), 2 ** (n - i))
|
34
|
-
|
35
|
-
# there are two ways to extract items :
|
36
|
-
# * NO_FALSE_POSITIVES includes all items with a lower bound
|
37
|
-
# above the a posteriori error
|
38
|
-
# * NO_FALSE_NEGATIVES includes all items with an uper bound
|
39
|
-
# above the a posteriori error
|
40
|
-
# a more complete discussion may be found at
|
41
|
-
# https://datasketches.github.io/docs/Frequency/FrequentItemsOverview.html
|
42
|
-
items_no_fp = fi.get_frequent_items(frequent_items_error_type.NO_FALSE_POSITIVES)
|
43
|
-
items_no_fn = fi.get_frequent_items(frequent_items_error_type.NO_FALSE_NEGATIVES)
|
44
|
-
self.assertLessEqual(len(items_no_fp), len(items_no_fn))
|
45
|
-
|
46
|
-
# the items list returns a decreasing weight-sorted list, and
|
47
|
-
# for each item we have (item, estimate, lower_bound, upper_bound)
|
48
|
-
item = items_no_fp[1]
|
49
|
-
self.assertLessEqual(item[2], item[1]) # lower bound vs estimate
|
50
|
-
self.assertLessEqual(item[1], item[3]) # estimate vs upper bound
|
51
|
-
|
52
|
-
# we can also query directly for a specific item
|
53
|
-
id = items_no_fn[0][0]
|
54
|
-
est = fi.get_estimate(id)
|
55
|
-
lb = fi.get_lower_bound(id)
|
56
|
-
ub = fi.get_upper_bound(id)
|
57
|
-
self.assertLessEqual(lb, est)
|
58
|
-
self.assertLessEqual(est, ub)
|
59
|
-
|
60
|
-
# the values are zero if the item isn't in our list
|
61
|
-
self.assertEqual(fi.get_estimate("NaN"), 0)
|
62
|
-
|
63
|
-
# now create a second sketch with a lot of unique
|
64
|
-
# values but all with equal weight (of 1) such that
|
65
|
-
# the total weight is much larger than the first sketch
|
66
|
-
fi2 = frequent_strings_sketch(k)
|
67
|
-
wt = fi.get_total_weight()
|
68
|
-
for i in range(0, 4*wt):
|
69
|
-
fi2.update(str(i))
|
70
|
-
|
71
|
-
# merge the second sketch into the first
|
72
|
-
fi.merge(fi2)
|
73
|
-
|
74
|
-
# we can see that the weight is much larger
|
75
|
-
self.assertEqual(5 * wt, fi.get_total_weight())
|
76
|
-
|
77
|
-
# querying with NO_FALSE_POSITIVES means we don't find anything
|
78
|
-
# heavy enough to return
|
79
|
-
items_no_fp = fi.get_frequent_items(frequent_items_error_type.NO_FALSE_POSITIVES)
|
80
|
-
self.assertEqual(len(items_no_fp), 0)
|
81
|
-
|
82
|
-
# we do, however, find a few potential heavy items
|
83
|
-
# if querying with NO_FALSE_NEGATIVES
|
84
|
-
items_no_fn = fi.get_frequent_items(frequent_items_error_type.NO_FALSE_NEGATIVES)
|
85
|
-
self.assertGreater(len(items_no_fn), 0)
|
86
|
-
|
87
|
-
# finally, serialize and reconstruct
|
88
|
-
fi_bytes = fi.serialize()
|
89
|
-
self.assertEqual(len(fi_bytes), fi.get_serialized_size_bytes())
|
90
|
-
new_fi = frequent_strings_sketch.deserialize(fi_bytes)
|
91
|
-
|
92
|
-
# and now interrogate the sketch
|
93
|
-
self.assertFalse(new_fi.is_empty())
|
94
|
-
self.assertGreater(new_fi.get_num_active_items(), 0)
|
95
|
-
self.assertEqual(5 * wt, new_fi.get_total_weight())
|
96
|
-
|
97
|
-
# This example uses generic objects but is otherwise identical
|
98
|
-
def test_fi_items_example(self):
|
99
|
-
k = 3 # a small value so we can easily fill the sketch
|
100
|
-
fi = frequent_items_sketch(k)
|
101
|
-
|
102
|
-
# as above, but in this case inserting ints
|
103
|
-
n = 8
|
104
|
-
for i in range(0, n):
|
105
|
-
fi.update(i, 2 ** (n - i))
|
106
|
-
|
107
|
-
# everything else works identically, so let's jump straight
|
108
|
-
# to merging and serialization
|
109
|
-
|
110
|
-
# now create a second sketch with a lot of unique
|
111
|
-
# values but all with equal weight (of 1) such that
|
112
|
-
# the total weight is much larger than the first sketch
|
113
|
-
fi2 = frequent_items_sketch(k)
|
114
|
-
wt = fi.get_total_weight()
|
115
|
-
for i in range(0, 4*wt):
|
116
|
-
fi2.update(i)
|
117
|
-
|
118
|
-
# merge the second sketch into the first
|
119
|
-
fi.merge(fi2)
|
120
|
-
|
121
|
-
# we can see that the weight is much larger
|
122
|
-
self.assertEqual(5 * wt, fi.get_total_weight())
|
123
|
-
|
124
|
-
# finally, serialize and reconstruct -- now we need a serde to tell
|
125
|
-
# (de)serialization how to interpret the objects
|
126
|
-
fi_bytes = fi.serialize(PyIntsSerDe())
|
127
|
-
self.assertEqual(len(fi_bytes), fi.get_serialized_size_bytes(PyIntsSerDe()))
|
128
|
-
new_fi = frequent_items_sketch.deserialize(fi_bytes, PyIntsSerDe())
|
129
|
-
|
130
|
-
# and again interrogate the sketch to check that it's what we serialized
|
131
|
-
self.assertFalse(new_fi.is_empty())
|
132
|
-
self.assertGreater(new_fi.get_num_active_items(), 0)
|
133
|
-
self.assertEqual(5 * wt, new_fi.get_total_weight())
|
134
|
-
|
135
|
-
|
136
|
-
def test_fi_sketch(self):
|
137
|
-
# only testing a few things not used in the above example
|
138
|
-
k = 12
|
139
|
-
wt = 10000
|
140
|
-
fi = frequent_strings_sketch(k)
|
141
|
-
|
142
|
-
self.assertAlmostEqual(fi.get_sketch_epsilon(), 0.0008545, delta=1e-6)
|
143
|
-
|
144
|
-
sk_apriori_error = fi.get_sketch_epsilon() * wt
|
145
|
-
reference_apriori_error = frequent_strings_sketch.get_apriori_error(k, wt)
|
146
|
-
self.assertAlmostEqual(sk_apriori_error, reference_apriori_error, delta=1e-6)
|
147
|
-
|
148
|
-
if __name__ == '__main__':
|
149
|
-
unittest.main()
|
@@ -1,129 +0,0 @@
|
|
1
|
-
# Licensed to the Apache Software Foundation (ASF) under one
|
2
|
-
# or more contributor license agreements. See the NOTICE file
|
3
|
-
# distributed with this work for additional information
|
4
|
-
# regarding copyright ownership. The ASF licenses this file
|
5
|
-
# to you under the Apache License, Version 2.0 (the
|
6
|
-
# "License"); you may not use this file except in compliance
|
7
|
-
# with the License. You may obtain a copy of the License at
|
8
|
-
#
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
10
|
-
#
|
11
|
-
# Unless required by applicable law or agreed to in writing,
|
12
|
-
# software distributed under the License is distributed on an
|
13
|
-
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
14
|
-
# KIND, either express or implied. See the License for the
|
15
|
-
# specific language governing permissions and limitations
|
16
|
-
# under the License.
|
17
|
-
|
18
|
-
import unittest
|
19
|
-
from datasketches import hll_sketch, hll_union, tgt_hll_type
|
20
|
-
|
21
|
-
class HllTest(unittest.TestCase):
|
22
|
-
def test_hll_example(self):
|
23
|
-
lgk = 12 # 2^k = 4096 rows in the table
|
24
|
-
n = 1 << 18 # ~256k unique values
|
25
|
-
|
26
|
-
# create a couple sketches and inject some values
|
27
|
-
# we'll have 1/4 of the values overlap
|
28
|
-
hll = hll_sketch(lgk, tgt_hll_type.HLL_8)
|
29
|
-
hll2 = hll_sketch(lgk, tgt_hll_type.HLL_6)
|
30
|
-
offset = int(3 * n / 4) # it's a float w/o cast
|
31
|
-
# because we hash on the bits, not an abstract numeric value,
|
32
|
-
# hll.update(1) and hll.update(1.0) give different results.
|
33
|
-
for i in range(0, n):
|
34
|
-
hll.update(i)
|
35
|
-
hll2.update(i + offset)
|
36
|
-
|
37
|
-
# we can check that the upper and lower bounds bracket the
|
38
|
-
# estimate, without needing to know the exact value.
|
39
|
-
self.assertLessEqual(hll.get_lower_bound(1), hll.get_estimate())
|
40
|
-
self.assertGreaterEqual(hll.get_upper_bound(1), hll.get_estimate())
|
41
|
-
|
42
|
-
# union is a separate class, and we can either get a result
|
43
|
-
# sketch or query the union object directly
|
44
|
-
union = hll_union(lgk)
|
45
|
-
union.update(hll)
|
46
|
-
union.update(hll2)
|
47
|
-
result = union.get_result()
|
48
|
-
self.assertEqual(result.get_estimate(), union.get_estimate())
|
49
|
-
|
50
|
-
# since our process here (including post-union HLL) is
|
51
|
-
# deterministic, we have checked and know the exact
|
52
|
-
# answer is within one standard deviation of the estimate
|
53
|
-
self.assertLessEqual(union.get_lower_bound(1), 7 * n / 4)
|
54
|
-
self.assertGreaterEqual(union.get_upper_bound(1), 7 * n / 4)
|
55
|
-
|
56
|
-
# serialize for storage and reconstruct
|
57
|
-
sk_bytes = result.serialize_compact()
|
58
|
-
self.assertEqual(len(sk_bytes), result.get_compact_serialization_bytes())
|
59
|
-
new_hll = hll_sketch.deserialize(sk_bytes)
|
60
|
-
|
61
|
-
# the sketch can self-report its configuration and status
|
62
|
-
self.assertEqual(new_hll.lg_config_k, lgk)
|
63
|
-
self.assertEqual(new_hll.tgt_type, tgt_hll_type.HLL_4)
|
64
|
-
self.assertFalse(new_hll.is_empty())
|
65
|
-
|
66
|
-
# if we want to reduce some object overhead, we can also reset
|
67
|
-
new_hll.reset()
|
68
|
-
self.assertTrue(new_hll.is_empty())
|
69
|
-
|
70
|
-
def test_hll_sketch(self):
|
71
|
-
lgk = 8
|
72
|
-
n = 117
|
73
|
-
hll = self.generate_sketch(n, lgk, tgt_hll_type.HLL_6)
|
74
|
-
hll.update('string data')
|
75
|
-
hll.update(3.14159) # double data
|
76
|
-
|
77
|
-
self.assertLessEqual(hll.get_lower_bound(1), hll.get_estimate())
|
78
|
-
self.assertGreaterEqual(hll.get_upper_bound(1), hll.get_estimate())
|
79
|
-
|
80
|
-
self.assertEqual(hll.lg_config_k, lgk)
|
81
|
-
self.assertEqual(hll.tgt_type, tgt_hll_type.HLL_6)
|
82
|
-
|
83
|
-
bytes_compact = hll.serialize_compact()
|
84
|
-
bytes_update = hll.serialize_updatable()
|
85
|
-
self.assertEqual(len(bytes_compact), hll.get_compact_serialization_bytes())
|
86
|
-
self.assertEqual(len(bytes_update), hll.get_updatable_serialization_bytes())
|
87
|
-
|
88
|
-
self.assertFalse(hll.is_compact())
|
89
|
-
self.assertFalse(hll.is_empty())
|
90
|
-
|
91
|
-
self.assertTrue(isinstance(hll_sketch.deserialize(bytes_compact), hll_sketch))
|
92
|
-
self.assertTrue(isinstance(hll_sketch.deserialize(bytes_update), hll_sketch))
|
93
|
-
|
94
|
-
self.assertIsNotNone(hll_sketch.get_rel_err(True, False, 12, 1))
|
95
|
-
self.assertIsNotNone(hll_sketch.get_max_updatable_serialization_bytes(20, tgt_hll_type.HLL_6))
|
96
|
-
|
97
|
-
hll.reset()
|
98
|
-
self.assertTrue(hll.is_empty())
|
99
|
-
|
100
|
-
def test_hll_union(self):
|
101
|
-
lgk = 7
|
102
|
-
n = 53
|
103
|
-
union = hll_union(lgk)
|
104
|
-
|
105
|
-
sk = self.generate_sketch(n, lgk, tgt_hll_type.HLL_4, 0)
|
106
|
-
union.update(sk)
|
107
|
-
sk = self.generate_sketch(3 * n, lgk, tgt_hll_type.HLL_4, n)
|
108
|
-
union.update(sk)
|
109
|
-
union.update('string data')
|
110
|
-
union.update(1.4142136)
|
111
|
-
|
112
|
-
self.assertLessEqual(union.get_lower_bound(1), union.get_estimate())
|
113
|
-
self.assertGreaterEqual(union.get_upper_bound(1), union.get_estimate())
|
114
|
-
|
115
|
-
self.assertEqual(union.lg_config_k, lgk)
|
116
|
-
self.assertFalse(union.is_empty())
|
117
|
-
|
118
|
-
sk = union.get_result()
|
119
|
-
self.assertTrue(isinstance(sk, hll_sketch))
|
120
|
-
self.assertEqual(sk.tgt_type, tgt_hll_type.HLL_4)
|
121
|
-
|
122
|
-
def generate_sketch(self, n, lgk, sk_type=tgt_hll_type.HLL_4, st_idx=0):
|
123
|
-
sk = hll_sketch(lgk, sk_type)
|
124
|
-
for i in range(st_idx, st_idx + n):
|
125
|
-
sk.update(i)
|
126
|
-
return sk
|
127
|
-
|
128
|
-
if __name__ == '__main__':
|
129
|
-
unittest.main()
|
@@ -1,159 +0,0 @@
|
|
1
|
-
# Licensed to the Apache Software Foundation (ASF) under one
|
2
|
-
# or more contributor license agreements. See the NOTICE file
|
3
|
-
# distributed with this work for additional information
|
4
|
-
# regarding copyright ownership. The ASF licenses this file
|
5
|
-
# to you under the Apache License, Version 2.0 (the
|
6
|
-
# "License"); you may not use this file except in compliance
|
7
|
-
# with the License. You may obtain a copy of the License at
|
8
|
-
#
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
10
|
-
#
|
11
|
-
# Unless required by applicable law or agreed to in writing,
|
12
|
-
# software distributed under the License is distributed on an
|
13
|
-
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
14
|
-
# KIND, either express or implied. See the License for the
|
15
|
-
# specific language governing permissions and limitations
|
16
|
-
# under the License.
|
17
|
-
|
18
|
-
import unittest
|
19
|
-
from datasketches import kll_ints_sketch, kll_floats_sketch, kll_doubles_sketch
|
20
|
-
from datasketches import kll_items_sketch, ks_test, PyStringsSerDe
|
21
|
-
import numpy as np
|
22
|
-
|
23
|
-
class KllTest(unittest.TestCase):
|
24
|
-
def test_kll_floats_example(self):
|
25
|
-
k = 160
|
26
|
-
n = 2 ** 20
|
27
|
-
|
28
|
-
# create a sketch and inject ~1 million N(0,1) points as an array and as a single item
|
29
|
-
kll = kll_floats_sketch(k)
|
30
|
-
kll.update(np.random.normal(size=n-1))
|
31
|
-
kll.update(0.0)
|
32
|
-
|
33
|
-
# 0 should be near the median
|
34
|
-
self.assertAlmostEqual(0.5, kll.get_rank(0.0), delta=0.035)
|
35
|
-
|
36
|
-
# the median should be near 0
|
37
|
-
self.assertAlmostEqual(0.0, kll.get_quantile(0.5), delta=0.035)
|
38
|
-
|
39
|
-
# we also track the min/max independently from the rest of the data
|
40
|
-
# which lets us know the full observed data range
|
41
|
-
self.assertLessEqual(kll.get_min_value(), kll.get_quantile(0.01))
|
42
|
-
self.assertLessEqual(0.0, kll.get_rank(kll.get_min_value()))
|
43
|
-
self.assertGreaterEqual(kll.get_max_value(), kll.get_quantile(0.99))
|
44
|
-
self.assertGreaterEqual(1.0, kll.get_rank(kll.get_max_value()))
|
45
|
-
|
46
|
-
# we can also extract a list of values at a time,
|
47
|
-
# here the values should give us something close to [-2, -1, 0, 1, 2].
|
48
|
-
# then get the CDF, which will return something close to
|
49
|
-
# the original values used in get_quantiles()
|
50
|
-
# finally, can check the normalized rank error bound
|
51
|
-
pts = kll.get_quantiles([0.0228, 0.1587, 0.5, 0.8413, 0.9772])
|
52
|
-
cdf = kll.get_cdf(pts) # include 1.0 at end to account for all probability mass
|
53
|
-
self.assertEqual(len(cdf), len(pts)+1)
|
54
|
-
err = kll.normalized_rank_error(False)
|
55
|
-
self.assertEqual(err, kll_floats_sketch.get_normalized_rank_error(k, False))
|
56
|
-
|
57
|
-
# and a few basic queries about the sketch
|
58
|
-
self.assertFalse(kll.is_empty())
|
59
|
-
self.assertTrue(kll.is_estimation_mode())
|
60
|
-
self.assertEqual(kll.get_n(), n)
|
61
|
-
self.assertEqual(kll.get_k(), k)
|
62
|
-
self.assertLess(kll.get_num_retained(), n)
|
63
|
-
|
64
|
-
# merging itself will double the number of items the sketch has seen
|
65
|
-
# but need to do that with a copy
|
66
|
-
kll_copy = kll_floats_sketch(kll)
|
67
|
-
kll.merge(kll_copy)
|
68
|
-
self.assertEqual(kll.get_n(), 2*n)
|
69
|
-
|
70
|
-
# we can then serialize and reconstruct the sketch
|
71
|
-
kll_bytes = kll.serialize()
|
72
|
-
new_kll = kll_floats_sketch.deserialize(kll_bytes)
|
73
|
-
self.assertEqual(kll.get_num_retained(), new_kll.get_num_retained())
|
74
|
-
self.assertEqual(kll.get_min_value(), new_kll.get_min_value())
|
75
|
-
self.assertEqual(kll.get_max_value(), new_kll.get_max_value())
|
76
|
-
self.assertEqual(kll.get_quantile(0.7), new_kll.get_quantile(0.7))
|
77
|
-
self.assertEqual(kll.get_rank(0.0), new_kll.get_rank(0.0))
|
78
|
-
|
79
|
-
# A Kolmogorov-Smirnov Test of kll and new_kll should match, even for
|
80
|
-
# a fairly small p-value -- cannot reject the null hypothesis that
|
81
|
-
# they come from the same distribution (since they do)
|
82
|
-
self.assertFalse(ks_test(kll, new_kll, 0.001))
|
83
|
-
|
84
|
-
total_weight = 0
|
85
|
-
for tuple in kll:
|
86
|
-
item = tuple[0]
|
87
|
-
weight = tuple[1]
|
88
|
-
total_weight = total_weight + weight
|
89
|
-
self.assertEqual(total_weight, kll.get_n())
|
90
|
-
|
91
|
-
def test_kll_ints_sketch(self):
|
92
|
-
k = 100
|
93
|
-
n = 10
|
94
|
-
kll = kll_ints_sketch(k)
|
95
|
-
for i in range(0, n):
|
96
|
-
kll.update(i)
|
97
|
-
|
98
|
-
self.assertEqual(kll.get_min_value(), 0)
|
99
|
-
self.assertEqual(kll.get_max_value(), n-1)
|
100
|
-
self.assertEqual(kll.get_n(), n)
|
101
|
-
self.assertFalse(kll.is_empty())
|
102
|
-
self.assertFalse(kll.is_estimation_mode()) # n < k
|
103
|
-
self.assertEqual(kll.get_k(), k)
|
104
|
-
|
105
|
-
pmf = kll.get_pmf([round(n/2)])
|
106
|
-
self.assertIsNotNone(pmf)
|
107
|
-
self.assertEqual(len(pmf), 2)
|
108
|
-
|
109
|
-
cdf = kll.get_cdf([round(n/2)])
|
110
|
-
self.assertIsNotNone(cdf)
|
111
|
-
self.assertEqual(len(cdf), 2)
|
112
|
-
|
113
|
-
self.assertEqual(kll.get_quantile(0.5), round(n/2))
|
114
|
-
quants = kll.get_quantiles([0.25, 0.5, 0.75])
|
115
|
-
self.assertIsNotNone(quants)
|
116
|
-
self.assertEqual(len(quants), 3)
|
117
|
-
|
118
|
-
self.assertEqual(kll.get_rank(round(n/2)), 0.5)
|
119
|
-
|
120
|
-
# merge copy of self
|
121
|
-
kll_copy = kll_ints_sketch(kll)
|
122
|
-
kll.merge(kll_copy)
|
123
|
-
self.assertEqual(kll.get_n(), 2 * n)
|
124
|
-
|
125
|
-
sk_bytes = kll.serialize()
|
126
|
-
self.assertTrue(isinstance(kll_ints_sketch.deserialize(sk_bytes), kll_ints_sketch))
|
127
|
-
|
128
|
-
def test_kll_doubles_sketch(self):
|
129
|
-
# already tested float and ints and it's templatized, so just make sure it instantiates properly
|
130
|
-
k = 75
|
131
|
-
kll = kll_doubles_sketch(k)
|
132
|
-
self.assertTrue(kll.is_empty())
|
133
|
-
|
134
|
-
def test_kll_items_sketch(self):
|
135
|
-
# most functionality has been tested, but we need to ensure objects and sorting work
|
136
|
-
# as well as serialization
|
137
|
-
k = 100
|
138
|
-
n = 2 ** 16
|
139
|
-
|
140
|
-
# create a sketch and inject enough points to force compaction
|
141
|
-
kll = kll_items_sketch(k)
|
142
|
-
for i in range(0, n):
|
143
|
-
kll.update(str(i))
|
144
|
-
|
145
|
-
kll_copy = kll_items_sketch(kll)
|
146
|
-
kll.merge(kll_copy)
|
147
|
-
self.assertEqual(kll.get_n(), 2 * n)
|
148
|
-
|
149
|
-
kll_bytes = kll.serialize(PyStringsSerDe())
|
150
|
-
new_kll = kll_items_sketch.deserialize(kll_bytes, PyStringsSerDe())
|
151
|
-
self.assertEqual(kll.get_num_retained(), new_kll.get_num_retained())
|
152
|
-
self.assertEqual(kll.get_min_value(), new_kll.get_min_value())
|
153
|
-
self.assertEqual(kll.get_max_value(), new_kll.get_max_value())
|
154
|
-
self.assertEqual(kll.get_quantile(0.7), new_kll.get_quantile(0.7))
|
155
|
-
self.assertEqual(kll.get_rank(str(n/4)), new_kll.get_rank(str(n/4)))
|
156
|
-
|
157
|
-
|
158
|
-
if __name__ == '__main__':
|
159
|
-
unittest.main()
|