mirror of
https://github.com/gsi-upm/senpy
synced 2024-11-22 16:12:29 +00:00
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
import os
|
||
|
|
import logging
|
||
|
|
logging.basicConfig()
|
||
|
|
try:
|
||
|
|
import unittest.mock as mock
|
||
|
|
except ImportError:
|
||
|
|
import mock
|
||
|
|
from senpy.extensions import Senpy
|
||
|
|
from flask import Flask
|
||
|
|
from flask.ext.testing import TestCase
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
class vaderTest(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.app = Flask("test_plugin")
|
||
|
|
self.dir = os.path.join(os.path.dirname(__file__))
|
||
|
|
self.senpy = Senpy(plugin_folder=self.dir, default_plugins=False)
|
||
|
|
self.senpy.init_app(self.app)
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
self.senpy.deactivate_plugin("vaderSentiment", sync=True)
|
||
|
|
|
||
|
|
def test_analyse(self):
|
||
|
|
plugin = self.senpy.plugins["vaderSentiment"]
|
||
|
|
plugin.activate()
|
||
|
|
|
||
|
|
texts = {'I am tired :(' : 'marl:Negative',
|
||
|
|
'I love pizza' : 'marl:Positive',
|
||
|
|
'I like going to the cinema :)' : 'marl:Positive',
|
||
|
|
'This cake is disgusting' : 'marl:Negative'}
|
||
|
|
|
||
|
|
for text in texts:
|
||
|
|
response = plugin.analyse(input=text)
|
||
|
|
expected = texts[text]
|
||
|
|
sentimentSet = response.entries[0].sentiments
|
||
|
|
|
||
|
|
max_sentiment = max(sentimentSet, key=lambda x: x['marl:polarityValue'])
|
||
|
|
assert max_sentiment['marl:hasPolarity'] == expected
|
||
|
|
|
||
|
|
plugin.deactivate()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|