| 123 | self.assertEqual(s(v).assessments[1], ([":-)"], +1.0, 1.0)) |
| 124 | |
| 125 | def test_bag_of_words(self): |
| 126 | # Assert weighted average polarity and subjectivity for bag-of-words with weighted features. |
| 127 | from pattern.vector import BagOfWords # Alias for pattern.vector.Document. |
| 128 | s = text.Sentiment() |
| 129 | v = BagOfWords({":-(": 3, ":-)": 1}) |
| 130 | self.assertEqual(s(v)[0], -0.5) |
| 131 | self.assertEqual(s(v)[1], +1.0) |
| 132 | self.assertEqual(s(v).assessments[0], ([":-("], -1.0, 1.0)) |
| 133 | self.assertEqual(s(v).assessments[1], ([":-)"], +1.0, 1.0)) |
| 134 | |
| 135 | #--------------------------------------------------------------------------------------------------- |
| 136 | |