def test_merge():
countmin1 = CountMinSketch(0.1, 0.1)
countmin2 = CountMinSketch(0.1, 0.1)
word = 'hello'
countmin1.increment(word)
countmin2.increment(word)
countmin1.merge(countmin2)
assert countmin2.estimate(word) == 1
assert countmin1.estimate(word) == 2
top_10_list = [('hello', 2.0)]
assert set(countmin1.top_10_dict.items()) == set(top_10_list)
def test_increment_and_estimate():
word1 = 'hello'
word2 = 'world'
word3 = 'other'
countmin = CountMinSketch(0.1, 0.1)
countmin.increment(word1)
countmin.increment(word2)
countmin.increment(word2)
assert countmin.estimate(word3) == 0
assert countmin.estimate(word1) == 1
assert countmin.estimate(word2) == 2
top_10_list = [('hello', 1.0), ('world', 2.0)]
assert set(countmin.top_10_dict.items()) == set(top_10_list)
