def test_get_all_cats(self): new_ids = ["foo", "bar", "baz"] for new_id in new_ids: self.r.hset( Category.CAT_META_HASH, "%s:signature" % (new_id), self.cat_sig ) cats = Category.get_all_categories(self.r) self.assertEqual(len(cats), len(new_ids) + 1) for cat in cats: self.assertEqual(cat.signature, self.cat_sig)
def create_mock_category(id, sig, thresh, name): """Create a mock category object for testing purposes. Args: id: str, system referenc (hash when not a test obj). sig: str, the signature we're matching against. thresh: float, what ratio will match sig. name: str, human-readable name. """ cat = Category( redis_util.MockRedis(share_state=False), cat_id=id) cat._set_signature(sig) cat._set_threshold(thresh) cat._set_human(name) return cat
def test_category_create(self): cat = Category.create(self.r, "new test category") self.assertEqual(cat.signature, "new test category") self.assertTrue(cat.cat_id)