1
0
mirror of https://github.com/gsi-upm/senpy synced 2024-11-21 07:42:28 +00:00

update to pass tests with community plugins

This commit is contained in:
J. Fernando Sánchez 2023-09-22 23:28:19 +02:00
parent 5b28b6d1b4
commit 4f95fbcbd1
15 changed files with 206 additions and 78 deletions

View File

@ -4,6 +4,10 @@ All notable changes to this project will be documented in this file.
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html). and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
## [Unreleased]
### Added
* The code of many senpy community plugins have been included by default. However, additional files (e.g., licensed data) and/or installing additional dependencies may be necessary for some plugins. Read each plugin's documentation for more information.
## [1.0.6] ## [1.0.6]
### Fixed ### Fixed
* Plugins now get activated for testing * Plugins now get activated for testing

View File

@ -19,7 +19,7 @@ COPY . /usr/src/app/
RUN pip install --no-cache-dir --no-index --no-deps --editable . RUN pip install --no-cache-dir --no-index --no-deps --editable .
ONBUILD COPY . /senpy-plugins/ ONBUILD COPY . /senpy-plugins/
ONBUILD RUN python -m senpy --only-install -f /senpy-plugins ONBUILD RUN python -m senpy -i --no-run -f /senpy-plugins
ONBUILD WORKDIR /senpy-plugins/ ONBUILD WORKDIR /senpy-plugins/
ENTRYPOINT ["python", "-m", "senpy", "-f", "/senpy-plugins/", "--host", "0.0.0.0"] ENTRYPOINT ["python", "-m", "senpy", "-f", "/senpy-plugins/", "--host", "0.0.0.0"]

View File

@ -10,8 +10,8 @@ The senpy server is launched via the `senpy` command:
usage: senpy [-h] [--level logging_level] [--log-format log_format] [--debug] usage: senpy [-h] [--level logging_level] [--log-format log_format] [--debug]
[--no-default-plugins] [--host HOST] [--port PORT] [--no-default-plugins] [--host HOST] [--port PORT]
[--plugins-folder PLUGINS_FOLDER] [--only-install] [--only-test] [--plugins-folder PLUGINS_FOLDER] [--install]
[--test] [--only-list] [--data-folder DATA_FOLDER] [--test] [--no-run] [--data-folder DATA_FOLDER]
[--no-threaded] [--no-deps] [--version] [--allow-fail] [--no-threaded] [--no-deps] [--version] [--allow-fail]
Run a Senpy server Run a Senpy server
@ -28,10 +28,9 @@ The senpy server is launched via the `senpy` command:
--port PORT, -p PORT Port to listen on. --port PORT, -p PORT Port to listen on.
--plugins-folder PLUGINS_FOLDER, -f PLUGINS_FOLDER --plugins-folder PLUGINS_FOLDER, -f PLUGINS_FOLDER
Where to look for plugins. Where to look for plugins.
--only-install, -i Do not run a server, only install plugin dependencies --install, -i Install plugin dependencies before launching the server.
--only-test Do not run a server, just test all plugins
--test, -t Test all plugins before launching the server --test, -t Test all plugins before launching the server
--only-list, --list Do not run a server, only list plugins found --no-run Do not launch the server
--data-folder DATA_FOLDER, --data DATA_FOLDER --data-folder DATA_FOLDER, --data DATA_FOLDER
Where to look for data. It be set with the SENPY_DATA Where to look for data. It be set with the SENPY_DATA
environment variable as well. environment variable as well.

View File

@ -31,7 +31,7 @@ pipeline = Pipeline([('cv', count_vec),
('clf', clf3)]) ('clf', clf3)])
pipeline.fit(X_train, y_train) pipeline.fit(X_train, y_train)
print('Feature names: {}'.format(count_vec.get_feature_names())) print('Feature names: {}'.format(count_vec.get_feature_names_out()))
print('Class count: {}'.format(clf3.class_count_)) print('Class count: {}'.format(clf3.class_count_))

View File

@ -22,6 +22,7 @@ the server.
from flask import Flask from flask import Flask
from senpy.extensions import Senpy from senpy.extensions import Senpy
from senpy.utils import easy_test from senpy.utils import easy_test
from senpy.plugins import list_dependencies
import logging import logging
import os import os
@ -81,16 +82,21 @@ def main():
action='append', action='append',
help='Where to look for plugins.') help='Where to look for plugins.')
parser.add_argument( parser.add_argument(
'--only-install', '--install',
'-i', '-i',
action='store_true', action='store_true',
default=False, default=False,
help='Do not run a server, only install plugin dependencies') help='Install plugin dependencies before running.')
parser.add_argument( parser.add_argument(
'--only-test', '--dependencies',
action='store_true', action='store_true',
default=False, default=False,
help='Do not run a server, just test all plugins') help='List plugin dependencies')
parser.add_argument(
'--strict',
action='store_true',
default=False,
help='Fail if optional plugins cannot be loaded.')
parser.add_argument( parser.add_argument(
'--test', '--test',
'-t', '-t',
@ -98,11 +104,10 @@ def main():
default=False, default=False,
help='Test all plugins before launching the server') help='Test all plugins before launching the server')
parser.add_argument( parser.add_argument(
'--only-list', '--no-run',
'--list',
action='store_true', action='store_true',
default=False, default=False,
help='Do not run a server, only list plugins found') help='Do not launch the server.')
parser.add_argument( parser.add_argument(
'--data-folder', '--data-folder',
'--data', '--data',
@ -156,6 +161,8 @@ def main():
sp = Senpy(app, sp = Senpy(app,
plugin_folder=None, plugin_folder=None,
default_plugins=not args.no_default_plugins, default_plugins=not args.no_default_plugins,
install=args.install,
strict=args.strict,
data_folder=args.data_folder) data_folder=args.data_folder)
folders = list(args.plugins_folder) if args.plugins_folder else [] folders = list(args.plugins_folder) if args.plugins_folder else []
if not folders: if not folders:
@ -175,17 +182,43 @@ def main():
fpath, fpath,
maxname=maxname, maxname=maxname,
maxversion=maxversion)) maxversion=maxversion))
if args.only_list: if args.dependencies:
return print('Listing dependencies')
if not args.no_deps: missing = []
installed = []
for plug in sp.plugins(is_activated=False).values():
inst, miss, nltkres = list_dependencies(plug)
if not any([inst, miss, nltkres]):
continue
print(f'Plugin: {plug.id}')
for m in miss:
missing.append(f'{m} # {plug.id}')
for i in inst:
installed.append(f'{i} # {plug.id}')
if installed:
print('Installed packages:')
for i in installed:
print(f'\t{i}')
if missing:
print('Missing packages:')
for m in missing:
print(f'\t{m}')
if args.install:
sp.install_deps() sp.install_deps()
if args.only_install:
if args.test:
sp.activate_all(sync=True)
easy_test(sp.plugins(is_activated=True), debug=args.debug)
if args.no_run:
return return
sp.activate_all(allow_fail=args.allow_fail)
if args.test or args.only_test: sp.activate_all(sync=True)
easy_test(sp.plugins(), debug=args.debug) if sp.strict:
if args.only_test: inactive = sp.plugins(is_activated=False)
return assert not inactive
print('Senpy version {}'.format(senpy.__version__)) print('Senpy version {}'.format(senpy.__version__))
print('Server running on port %s:%d. Ctrl+C to quit' % (args.host, print('Server running on port %s:%d. Ctrl+C to quit' % (args.host,
args.port)) args.port))

View File

@ -44,6 +44,8 @@ class Senpy(object):
app=None, app=None,
plugin_folder=".", plugin_folder=".",
data_folder=None, data_folder=None,
install=False,
strict=True,
default_plugins=False): default_plugins=False):
default_data = os.path.join(os.getcwd(), 'senpy_data') default_data = os.path.join(os.getcwd(), 'senpy_data')
@ -57,6 +59,8 @@ class Senpy(object):
raise raise
self._default = None self._default = None
self.strict = strict
self.install = install
self._plugins = {} self._plugins = {}
if plugin_folder: if plugin_folder:
self.add_folder(plugin_folder) self.add_folder(plugin_folder)
@ -148,7 +152,8 @@ class Senpy(object):
logger.debug("Adding folder: %s", folder) logger.debug("Adding folder: %s", folder)
if os.path.isdir(folder): if os.path.isdir(folder):
new_plugins = plugins.from_folder([folder], new_plugins = plugins.from_folder([folder],
data_folder=self.data_folder) data_folder=self.data_folder,
strict=self.strict)
for plugin in new_plugins: for plugin in new_plugins:
self.add_plugin(plugin) self.add_plugin(plugin)
else: else:
@ -173,7 +178,7 @@ class Senpy(object):
logger.info('Installing dependencies') logger.info('Installing dependencies')
# If a plugin is activated, its dependencies should already be installed # If a plugin is activated, its dependencies should already be installed
# Otherwise, it would've failed to activate. # Otherwise, it would've failed to activate.
plugins.install_deps(*self.plugins(is_activated=False)) plugins.install_deps(*self._plugins.values())
def analyse(self, request, analyses=None): def analyse(self, request, analyses=None):
""" """
@ -340,13 +345,13 @@ class Senpy(object):
else: else:
self._default = self._plugins[value.lower()] self._default = self._plugins[value.lower()]
def activate_all(self, sync=True, allow_fail=False): def activate_all(self, sync=True):
ps = [] ps = []
for plug in self._plugins.keys(): for plug in self._plugins.keys():
try: try:
self.activate_plugin(plug, sync=sync) self.activate_plugin(plug, sync=sync)
except Exception as ex: except Exception as ex:
if not allow_fail: if self.strict:
raise raise
logger.error('Could not activate {}: {}'.format(plug, ex)) logger.error('Could not activate {}: {}'.format(plug, ex))
return ps return ps
@ -358,15 +363,19 @@ class Senpy(object):
return ps return ps
def _activate(self, plugin): def _activate(self, plugin):
success = False
with plugin._lock: with plugin._lock:
if plugin.is_activated: if plugin.is_activated:
logger.info(f"Plugin is already activated: {plugin.name}")
return return
plugin._activate() try:
msg = "Plugin activated: {}".format(plugin.name) assert plugin._activate()
logger.info(msg) logger.info(f"Plugin activated: {plugin.name}")
success = plugin.is_activated except Exception as ex:
return success if getattr(plugin, "optional", False) and not self.strict:
logger.info(f"Plugin could NOT be activated: {plugin.name}")
return False
raise
return plugin.is_activated
def activate_plugin(self, plugin_name, sync=True): def activate_plugin(self, plugin_name, sync=True):
plugin_name = plugin_name.lower() plugin_name = plugin_name.lower()

View File

@ -155,8 +155,11 @@ class Plugin(with_metaclass(PluginMeta, models.Plugin)):
return os.path.dirname(inspect.getfile(self.__class__)) return os.path.dirname(inspect.getfile(self.__class__))
def _activate(self): def _activate(self):
if self.is_activated:
return
self.activate() self.activate()
self.is_activated = True self.is_activated = True
return self.is_activated
def _deactivate(self): def _deactivate(self):
self.is_activated = False self.is_activated = False
@ -262,11 +265,13 @@ class Plugin(with_metaclass(PluginMeta, models.Plugin)):
assert not should_fail assert not should_fail
def find_file(self, fname): def find_file(self, fname):
tried = []
for p in self._data_paths: for p in self._data_paths:
alternative = os.path.join(p, fname) alternative = os.path.abspath(os.path.join(p, fname))
if os.path.exists(alternative): if os.path.exists(alternative):
return alternative return alternative
raise IOError('File does not exist: {}'.format(fname)) tried.append(alternative)
raise IOError(f'File does not exist: {fname}. Tried: {tried}')
def path(self, fpath): def path(self, fpath):
if not os.path.isabs(fpath): if not os.path.isabs(fpath):
@ -290,6 +295,20 @@ class Plugin(with_metaclass(PluginMeta, models.Plugin)):
SenpyPlugin = Plugin SenpyPlugin = Plugin
class FailedPlugin(Plugin):
"""A plugin that has failed to initialize."""
version = 0
def __init__(self, info, function):
super().__init__(info)
a = info.get('name', info.get('module', self.name))
self['name'] == a
self._function = function
def retry(self):
return self._function()
class Analyser(Plugin): class Analyser(Plugin):
''' '''
A subclass of Plugin that analyses text and provides an annotation. A subclass of Plugin that analyses text and provides an annotation.
@ -699,23 +718,31 @@ def missing_requirements(reqs):
res = pool.apply_async(pkg_resources.get_distribution, (req,)) res = pool.apply_async(pkg_resources.get_distribution, (req,))
queue.append((req, res)) queue.append((req, res))
missing = [] missing = []
installed = []
for req, job in queue: for req, job in queue:
try: try:
job.get(1) installed.append(job.get(1))
except Exception: except Exception:
missing.append(req) missing.append(req)
return missing return installed, missing
def list_dependencies(*plugins):
'''List all dependencies (python and nltk) for the given list of plugins'''
nltk_resources = set()
missing = []
installed = []
for info in plugins:
reqs = info.get('requirements', [])
if reqs:
inst, miss= missing_requirements(reqs)
installed += inst
missing += miss
nltk_resources |= set(info.get('nltk_resources', []))
return installed, missing, nltk_resources
def install_deps(*plugins): def install_deps(*plugins):
_, requirements, nltk_resources = list_dependencies(*plugins)
installed = False installed = False
nltk_resources = set()
requirements = []
for info in plugins:
requirements = info.get('requirements', [])
if requirements:
requirements += missing_requirements(requirements)
nltk_resources |= set(info.get('nltk_resources', []))
if requirements: if requirements:
logger.info('Installing requirements: ' + str(requirements)) logger.info('Installing requirements: ' + str(requirements))
pip_args = [sys.executable, '-m', 'pip', 'install'] pip_args = [sys.executable, '-m', 'pip', 'install']
@ -729,8 +756,7 @@ def install_deps(*plugins):
if exitcode != 0: if exitcode != 0:
raise models.Error( raise models.Error(
"Dependencies not properly installed: {}".format(pip_args)) "Dependencies not properly installed: {}".format(pip_args))
installed |= download(list(nltk_resources)) return installed or download(list(nltk_resources))
return installed
is_plugin_file = re.compile(r'.*\.senpy$|senpy_[a-zA-Z0-9_]+\.py$|' is_plugin_file = re.compile(r'.*\.senpy$|senpy_[a-zA-Z0-9_]+\.py$|'
@ -747,7 +773,7 @@ def find_plugins(folders):
yield fpath yield fpath
def from_path(fpath, install_on_fail=False, **kwargs): def from_path(fpath, **kwargs):
logger.debug("Loading plugin from {}".format(fpath)) logger.debug("Loading plugin from {}".format(fpath))
if fpath.endswith('.py'): if fpath.endswith('.py'):
# We asume root is the dir of the file, and module is the name of the file # We asume root is the dir of the file, and module is the name of the file
@ -757,18 +783,18 @@ def from_path(fpath, install_on_fail=False, **kwargs):
yield instance yield instance
else: else:
info = parse_plugin_info(fpath) info = parse_plugin_info(fpath)
yield from_info(info, install_on_fail=install_on_fail, **kwargs) yield from_info(info, **kwargs)
def from_folder(folders, loader=from_path, **kwargs): def from_folder(folders, loader=from_path, **kwargs):
plugins = [] plugins = []
for fpath in find_plugins(folders): for fpath in find_plugins(folders):
for plugin in loader(fpath, **kwargs): for plugin in loader(fpath, **kwargs):
plugins.append(plugin) if plugin:
plugins.append(plugin)
return plugins return plugins
def from_info(info, root=None, install_on_fail=True, **kwargs): def from_info(info, root=None, strict=False, **kwargs):
if any(x not in info for x in ('module', )): if any(x not in info for x in ('module', )):
raise ValueError('Plugin info is not valid: {}'.format(info)) raise ValueError('Plugin info is not valid: {}'.format(info))
module = info["module"] module = info["module"]
@ -780,8 +806,10 @@ def from_info(info, root=None, install_on_fail=True, **kwargs):
try: try:
return fun() return fun()
except (ImportError, LookupError): except (ImportError, LookupError):
install_deps(info) if strict or not str(info.get("optional", "false")).lower() in ["True", "true", "t"]:
return fun() raise
print(f"Could not import plugin: { info }")
return FailedPlugin(info, fun)
def parse_plugin_info(fpath): def parse_plugin_info(fpath):

View File

@ -23,6 +23,51 @@ from senpy.plugins import EmotionPlugin, SenpyPlugin
from senpy.models import Results, EmotionSet, Entry, Emotion from senpy.models import Results, EmotionSet, Entry, Emotion
### BEGIN WORKAROUND FOR PATTERN
# See: https://github.com/clips/pattern/issues/308
import os.path
import pattern.text
from pattern.helpers import decode_string
from codecs import BOM_UTF8
BOM_UTF8 = BOM_UTF8.decode("utf-8")
decode_utf8 = decode_string
MODEL = "emoml:pad-dimensions_"
VALENCE = f"{MODEL}_valence"
AROUSAL = f"{MODEL}_arousal"
DOMINANCE = f"{MODEL}_dominance"
def _read(path, encoding="utf-8", comment=";;;"):
"""Returns an iterator over the lines in the file at the given path,
strippping comments and decoding each line to Unicode.
"""
if path:
if isinstance(path, str) and os.path.exists(path):
# From file path.
f = open(path, "r", encoding="utf-8")
elif isinstance(path, str):
# From string.
f = path.splitlines()
else:
# From file or buffer.
f = path
for i, line in enumerate(f):
line = line.strip(BOM_UTF8) if i == 0 and isinstance(line, str) else line
line = line.strip()
line = decode_utf8(line, encoding)
if not line or (comment and line.startswith(comment)):
continue
yield line
pattern.text._read = _read
## END WORKAROUND
class ANEW(EmotionPlugin): class ANEW(EmotionPlugin):
description = "This plugin consists on an emotion classifier using ANEW lexicon dictionary. It averages the VAD (valence-arousal-dominance) value of each word in the text that is also in the ANEW dictionary. To obtain a categorical value (e.g., happy) use the emotion conversion API (e.g., `emotion-model=emoml:big6`)." description = "This plugin consists on an emotion classifier using ANEW lexicon dictionary. It averages the VAD (valence-arousal-dominance) value of each word in the text that is also in the ANEW dictionary. To obtain a categorical value (e.g., happy) use the emotion conversion API (e.g., `emotion-model=emoml:big6`)."
author = "@icorcuera" author = "@icorcuera"
@ -41,7 +86,7 @@ class ANEW(EmotionPlugin):
anew_path_es = "Dictionary/Redondo(2007).csv" anew_path_es = "Dictionary/Redondo(2007).csv"
anew_path_en = "Dictionary/ANEW2010All.txt" anew_path_en = "Dictionary/ANEW2010All.txt"
onyx__usesEmotionModel = "emoml:pad-dimensions" onyx__usesEmotionModel = MODEL
nltk_resources = ['stopwords'] nltk_resources = ['stopwords']
def activate(self, *args, **kwargs): def activate(self, *args, **kwargs):
@ -147,9 +192,9 @@ class ANEW(EmotionPlugin):
emotions.id = "Emotions0" emotions.id = "Emotions0"
emotion1 = Emotion(id="Emotion0") emotion1 = Emotion(id="Emotion0")
emotion1["emoml:pad-dimensions_pleasure"] = feature_set['V'] emotion1[VALENCE] = feature_set['V']
emotion1["emoml:pad-dimensions_arousal"] = feature_set['A'] emotion1[AROUSAL] = feature_set['A']
emotion1["emoml:pad-dimensions_dominance"] = feature_set['D'] emotion1[DOMINANCE] = feature_set['D']
emotion1.prov(activity) emotion1.prov(activity)
emotions.prov(activity) emotions.prov(activity)
@ -159,7 +204,6 @@ class ANEW(EmotionPlugin):
yield entry yield entry
ontology = "http://gsi.dit.upm.es/ontologies/wnaffect/ns#"
test_cases = [ test_cases = [
{ {
'name': 'anger with VAD=(2.12, 6.95, 5.05)', 'name': 'anger with VAD=(2.12, 6.95, 5.05)',
@ -167,9 +211,9 @@ class ANEW(EmotionPlugin):
'expected': { 'expected': {
'onyx:hasEmotionSet': [{ 'onyx:hasEmotionSet': [{
'onyx:hasEmotion': [{ 'onyx:hasEmotion': [{
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#arousal": 6.95, AROUSAL: 6.95,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#dominance": 5.05, DOMINANCE: 5.05,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#valence": 2.12, VALENCE: 2.12,
}] }]
}] }]
} }
@ -178,9 +222,7 @@ class ANEW(EmotionPlugin):
'expected': { 'expected': {
'onyx:hasEmotionSet': [{ 'onyx:hasEmotionSet': [{
'onyx:hasEmotion': [{ 'onyx:hasEmotion': [{
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#arousal": 4.13, f"{MODEL}_arousal": 4.13,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#dominance": 3.45,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#valence": 1.61,
}] }]
}] }]
@ -191,9 +233,9 @@ class ANEW(EmotionPlugin):
'expected': { 'expected': {
'onyx:hasEmotionSet': [{ 'onyx:hasEmotionSet': [{
'onyx:hasEmotion': [{ 'onyx:hasEmotion': [{
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#arousal": 6.49, AROUSAL: 6.49,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#dominance": 6.63, DOMINANCE: 6.63,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#valence": 8.21, VALENCE: 8.21,
}] }]
}] }]
} }
@ -203,9 +245,9 @@ class ANEW(EmotionPlugin):
'expected': { 'expected': {
'onyx:hasEmotionSet': [{ 'onyx:hasEmotionSet': [{
'onyx:hasEmotion': [{ 'onyx:hasEmotion': [{
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#arousal": 5.8100000000000005, AROUSAL: 5.8100000000000005,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#dominance": 4.33, DOMINANCE: 4.33,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#valence": 5.050000000000001, VALENCE: 5.050000000000001,
}] }]
}] }]
@ -216,9 +258,9 @@ class ANEW(EmotionPlugin):
'expected': { 'expected': {
'onyx:hasEmotionSet': [{ 'onyx:hasEmotionSet': [{
'onyx:hasEmotion': [{ 'onyx:hasEmotion': [{
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#arousal": 5.09, AROUSAL: 5.09,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#dominance": 4.4, DOMINANCE: 4.4,
"http://www.gsi.dit.upm.es/ontologies/onyx/vocabularies/anew/ns#valence": 5.109999999999999, VALENCE: 5.109999999999999,
}] }]
}] }]

View File

@ -1,5 +1,6 @@
--- ---
module: emotion-anew module: emotion-anew
optional: true
requirements: requirements:
- numpy - numpy
- pandas - pandas

View File

@ -33,6 +33,7 @@ class DepecheMood(EmotionBox):
name = 'emotion-depechemood' name = 'emotion-depechemood'
version = '0.1' version = '0.1'
requirements = ['pandas'] requirements = ['pandas']
optional = True
nltk_resources = ["stopwords"] nltk_resources = ["stopwords"]
onyx__usesEmotionModel = 'wna:WNAModel' onyx__usesEmotionModel = 'wna:WNAModel'

View File

@ -1,5 +1,6 @@
--- ---
module: emotion-wnaffect module: emotion-wnaffect
optional: true
requirements: requirements:
- nltk>=3.0.5 - nltk>=3.0.5
- lxml>=3.4.2 - lxml>=3.4.2

View File

@ -1,5 +1,6 @@
--- ---
module: sentiment-basic module: sentiment-basic
optional: true
requirements: requirements:
- nltk>=3.0.5 - nltk>=3.0.5
- scipy>=0.14.0 - scipy>=0.14.0

View File

@ -44,6 +44,10 @@ def check_template(indict, template):
raise models.Error(('Element not found.' raise models.Error(('Element not found.'
'\nExpected: {}\nIn: {}').format(pprint.pformat(e), '\nExpected: {}\nIn: {}').format(pprint.pformat(e),
pprint.pformat(indict))) pprint.pformat(indict)))
elif isinstance(template, float) and isinstance(indict, float):
diff = abs(indict - template)
if (diff > 0) and diff/(abs(indict+template)) > 0.05:
raise models.Error('Differences greater than 10% found.\n')
else: else:
if indict != template: if indict != template:
raise models.Error(('Differences found.\n' raise models.Error(('Differences found.\n'

View File

@ -209,8 +209,8 @@ class BlueprintsTest(TestCase):
""" """
# First, we split by sentence twice. Each call should generate 3 additional entries # First, we split by sentence twice. Each call should generate 3 additional entries
# (one per sentence in the original). # (one per sentence in the original).
resp = self.client.get('/api/split/split?i=The first sentence. The second sentence.' resp = self.client.get('/api/split/split?i=The first sentence. The second sentence.%0A'
'\nA new paragraph&delimiter=sentence&verbose') 'A new paragraph&delimiter=sentence&verbose')
js = parse_resp(resp) js = parse_resp(resp)
assert len(js['activities']) == 2 assert len(js['activities']) == 2
assert len(js['entries']) == 7 assert len(js['entries']) == 7
@ -218,9 +218,8 @@ class BlueprintsTest(TestCase):
# Now, we split by sentence. This produces 3 additional entries. # Now, we split by sentence. This produces 3 additional entries.
# Then, we split by paragraph. This should create 2 additional entries (One per paragraph # Then, we split by paragraph. This should create 2 additional entries (One per paragraph
# in the original text) # in the original text)
resp = self.client.get('/api/split/split?i=The first sentence. The second sentence.' resp = self.client.get('/api/split/split?i=The first sentence. The second sentence.%0AA new paragraph'
'\nA new paragraph&0.delimiter=sentence' '&0.delimiter=sentence&1.delimiter=paragraph&verbose')
'&1.delimiter=paragraph&verbose')
# Calling dummy twice, should return the same string # Calling dummy twice, should return the same string
self.assertCode(resp, 200) self.assertCode(resp, 200)
js = parse_resp(resp) js = parse_resp(resp)

View File

@ -255,3 +255,9 @@ class ExtensionsTest(TestCase):
self.senpy.analyse(r3) self.senpy.analyse(r3)
assert len(r3.entries[0].emotions) == 1 assert len(r3.entries[0].emotions) == 1
r3.jsonld() r3.jsonld()
def testDefaultPlugins(self):
'''The default set of plugins should all load'''
self.app = Flask('test_extensions')
self.examples_dir = os.path.join(os.path.dirname(__file__), '..', 'example-plugins')
self.senpy = Senpy(app=self.app, default_plugins=False, strict=True)