sentimentpy模块进行中文文本情感分类

import unittest
from sentiment import sentiment

class Test(unittest.TestCase):

    def test_chinese(self):
        classifier = sentiment.Classifier('zh')
        # 必须用空格分词
        rtn = classifier.polarity('我 喜欢 这个 手机')
        self.assertEqual(rtn['best_fit_cat'], 'positive')
        rtn = classifier.polarity('我 今天 很 难过')
        self.assertEqual(rtn['best_fit_cat'], 'neutral')
        rtn = classifier.polarity('我 讨厌 你 你 滚开')
        self.assertEqual(rtn['best_fit_cat'], 'negative')

        rtn = classifier.emotion('我 喜欢 这个 手机')
        self.assertEqual(rtn['best_fit_cat'], '好')
        rtn = classifier.emotion('我 今天 很 难过')
        self.assertEqual(rtn['best_fit_cat'], '哀')
        rtn = classifier.emotion('我 讨厌 你 你 滚开')
        self.assertEqual(rtn['best_fit_cat'], '恶')

    
    def test_english(self):
        classifier = sentiment.Classifier('en')
        rtn = classifier.polarity('i love this cellphone')
        self.assertEqual(rtn['best_fit_cat'], 'positive')
        rtn = classifier.polarity('i am so sad today')
        self.assertEqual(rtn['best_fit_cat'], 'negative')
        rtn = classifier.polarity('i hate you fuck away')
        self.assertEqual(rtn['best_fit_cat'], 'negative')

        rtn = classifier.emotion('i love this cellphone')
        self.assertEqual(rtn['best_fit_cat'], 'joy')
        rtn = classifier.emotion('i am so sad today')
        self.assertEqual(rtn['best_fit_cat'], 'sadness')
        rtn = classifier.emotion('i hate you fuck away')
        self.assertEqual(rtn['best_fit_cat'], 'anger')

if __name__ == '__main__':
    unittest.main()