def test_consistent_merge():
""" Test that merge() calls do not modify the argument sketch.
"""
s1 = GKArray(test_eps)
s2 = GKArray(test_eps)
d = Normal(100)
for v in d.data:
s1.add(v)
s1.merge(s2)
# s2 is still empty
np.testing.assert_equal(s2.num_values, 0)
d = Normal(50)
for v in d.data:
s2.add(v)
s2_summary = [s2.quantile(q) for q in test_quantiles]+ [s2.sum, s2.avg, s2.num_values]
s1.merge(s2)
d = Normal(10)
for v in d.data:
s1.add(v)
# changes to s1 does not affect s2 after merge
s2_summary = [s2.quantile(q) for q in test_quantiles] + [s2.sum, s2.avg, s2.num_values]
np.testing.assert_almost_equal([s2.quantile(q) for q in test_quantiles] + [s2.sum, s2.avg, s2.num_values],
s2_summary)
s3 = GKArray(test_eps)
s3.merge(s2)
# merging to an empty sketch does not change s2
np.testing.assert_almost_equal([s2.quantile(q) for q in test_quantiles] + [s2.sum, s2.avg, s2.num_values],
s2_summary)
def test_constant():
for n in test_sizes:
data = Constant(n)
sketch = GKArray(test_eps)
for v in data.data:
sketch.add(v)
for q in test_quantiles:
np.testing.assert_equal(sketch.quantile(q), 42)