def test_word_exists_in_category_features(self):
c = Classify()
w = "crazyjackson"
cat = "gossip"
c.add_feature_to_category(category=cat, feature=w, weight=1)
r = c.word_exists_in_category_features(category=cat, w=w)
self.assertGreater(r, 0)
def test_get_category_for_text(self):
c = Classify()
ct = 'gossip'
f = 'lol'
ct1 = 'other'
f1 = 'random'
text = 'this guy is crazy ' + f
self.test_add_feature_to_category(category=ct, feature=f)
self.test_add_feature_to_category(category=ct1, feature=f1)
c.refresh_feature_set()
cat = c.get_category_for_text(text=text)
self.assertEqual(cat, ct)
def get(self):
try:
email = tornado.escape.xhtml_escape(self.current_user["email"])
if email not in settings.ADMIN_EMAILS:
self.render('denied.html')
c = Classify()
result = c.get_feature_set()
#self.write(json.dumps(result))
self.render("bo_category_features.html", categories=result)
except Exception,e:
logging.exception(e)
self.render("404.html")
def test_evaluate_category(self):
c = Classify()
cs = {"gossip":1, "science":3, "concert": 2}
res = c.evaluate_category(category_score=cs)
self.assertEqual(res, 'science')
def test_add_feature_to_category(self, category = 'gossip', feature='lol'):
c = Classify()
c.add_feature_to_category(category=category, feature=feature, weight=1)
r = c.get_feature_set()
self.assertNotEqual(r, {})
def test_get_feature_set(self):
c = Classify()
r = c.get_feature_set()
self.assertEqual(r, {})