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_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, {})
