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)