diff --git a/senpy/utils.py b/senpy/utils.py index 85fb5c3..1f14de3 100644 --- a/senpy/utils.py +++ b/senpy/utils.py @@ -71,8 +71,9 @@ def easy_test(plugin_list=None): logger.setLevel(logging.DEBUG) logging.getLogger().setLevel(logging.INFO) if not plugin_list: - from . import plugins import __main__ + logger.info('Loading classes from {}'.format(__main__)) + from . import plugins plugin_list = plugins.from_module(__main__) for plug in plugin_list: plug.test() diff --git a/tests/test_models.py b/tests/test_models.py index abdc8f0..0f8d170 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -9,6 +9,7 @@ from senpy.models import (Emotion, EmotionAnalysis, EmotionSet, Entry, + Entity, Error, Results, Sentiment, @@ -207,3 +208,14 @@ class ModelsTest(TestCase): recovered = from_string(string) assert isinstance(recovered, Results) assert isinstance(recovered.entries[0], Entry) + + def test_serializable(self): + r = Results() + e = Entry() + ent = Entity() + e.entities.append(ent) + r.entries.append(e) + d = r.serializable() + assert d + assert d['entries'] + assert d['entries'][0]['entities']