def test_init():
cms = CountMinSketch(2, 4)
assert cms.sizeof() == 32, 'Unexpected size in bytes'
with pytest.raises(ValueError) as excinfo:
cms = CountMinSketch(0, 5)
assert str(excinfo.value) == 'At least one counter array is required'
with pytest.raises(ValueError) as excinfo:
cms = CountMinSketch(5, 0)
assert str(excinfo.value) == (
'The length of the counter array cannot be less then 1')
21,
9,
29,
6,
5,
2,
3,
1,
16,
17,
15,
5,
3,
6,
9,
12,
]
if __name__ == '__main__':
cms = CountMinSketch(4, 100)
print(cms)
print("CMS uses {} bytes in the memory".format(cms.sizeof()))
for digit in DATASET:
cms.add(digit)
for digit in sorted(set(DATASET)):
print("Element: {}. Freq.: {}, Est. Freq.: {}".format(
digit, DATASET.count(digit), cms.frequency(digit)))
def test_size():
cms = CountMinSketch(2, 4)
element_size = array.array('I', [1]).itemsize
assert cms.sizeof() == element_size * len(cms), "Unexpected size in bytes"