def test_cms_set_query_type(self):
"""test setting different query types"""
cms = CountMinSketch(width=1000, depth=5)
self.assertEqual(cms.query_type, "min")
cms.query_type = "mean-min"
self.assertEqual(cms.query_type, "mean-min")
cms.query_type = "mean"
self.assertEqual(cms.query_type, "mean")
cms.query_type = "unknown"
self.assertEqual(cms.query_type, "min")
def test_cms_set_query_type(self):
''' test setting different query types '''
cms = CountMinSketch(width=1000, depth=5)
self.assertEqual(cms.query_type, 'min')
cms.query_type = 'mean-min'
self.assertEqual(cms.query_type, 'mean-min')
cms.query_type = 'mean'
self.assertEqual(cms.query_type, 'mean')
cms.query_type = 'unknown'
self.assertEqual(cms.query_type, 'min')
def test_cms_set_query_type(self):
''' test setting different query types '''
cms = CountMinSketch(width=1000, depth=5)
self.assertEqual(cms.query_type, 'min')
cms.query_type = 'mean-min'
self.assertEqual(cms.query_type, 'mean-min')
cms.query_type = 'mean'
self.assertEqual(cms.query_type, 'mean')
cms.query_type = 'unknown'
self.assertEqual(cms.query_type, 'min')
def test_cms_check_mean_called(self):
''' test checking number elements using mean algorithm called out '''
cms = CountMinSketch(width=1000, depth=5)
cms.query_type = 'mean'
self.assertEqual(cms.add('this is a test', 255), 255)
self.assertEqual(cms.add('this is another test', 189), 189)
self.assertEqual(cms.add('this is also a test', 16), 16)
self.assertEqual(cms.add('this is something to test', 5), 5)
self.assertEqual(cms.check('this is something to test'), 5)
self.assertEqual(cms.check('this is also a test'), 16)
self.assertEqual(cms.check('this is another test'), 189)
self.assertEqual(cms.check('this is a test'), 255)
self.assertEqual(cms.elements_added, 5 + 16 + 189 + 255)
def test_cms_check_mean_called(self):
"""test checking number elements using mean algorithm called out"""
cms = CountMinSketch(width=1000, depth=5)
cms.query_type = "mean"
self.assertEqual(cms.add("this is a test", 255), 255)
self.assertEqual(cms.add("this is another test", 189), 189)
self.assertEqual(cms.add("this is also a test", 16), 16)
self.assertEqual(cms.add("this is something to test", 5), 5)
self.assertEqual(cms.check("this is something to test"), 5)
self.assertEqual(cms.check("this is also a test"), 16)
self.assertEqual(cms.check("this is another test"), 189)
self.assertEqual(cms.check("this is a test"), 255)
self.assertEqual(cms.elements_added, 5 + 16 + 189 + 255)
def test_cms_check_min_called(self):
''' test checking number elements using min algorithm called out '''
cms = CountMinSketch(width=1000, depth=5)
cms.query_type = None
self.assertEqual(cms.add('this is a test', 255), 255)
self.assertEqual(cms.add('this is another test', 189), 189)
self.assertEqual(cms.add('this is also a test', 16), 16)
self.assertEqual(cms.add('this is something to test', 5), 5)
self.assertEqual(cms.check('this is something to test'), 5)
self.assertEqual(cms.check('this is also a test'), 16)
self.assertEqual(cms.check('this is another test'), 189)
self.assertEqual(cms.check('this is a test'), 255)
self.assertEqual(cms.elements_added, 5 + 16 + 189 + 255)
def test_cms_check_mean_called_even(self):
''' test checking number elements using mean algorithm called out when
the depth is an even number... '''
cms = CountMinSketch(width=1000, depth=6)
cms.query_type = 'mean-min'
self.assertEqual(cms.add('this is a test', 255), 255)
self.assertEqual(cms.add('this is another test', 189), 189)
self.assertEqual(cms.add('this is also a test', 16), 16)
self.assertEqual(cms.add('this is something to test', 5), 5)
self.assertEqual(cms.check('this is something to test'), 5)
self.assertEqual(cms.check('this is also a test'), 16)
self.assertEqual(cms.check('this is another test'), 189)
self.assertEqual(cms.check('this is a test'), 255)
self.assertEqual(cms.elements_added, 5 + 16 + 189 + 255)