From d29c42fd2e92fae611e33615ec28b0156c75c969 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?J=2E=20Fernando=20S=C3=A1nchez?= Date: Thu, 18 Jan 2018 15:50:46 +0100 Subject: [PATCH] Log easy and test serializable --- senpy/utils.py | 3 ++- tests/test_models.py | 12 ++++++++++++ 2 files changed, 14 insertions(+), 1 deletion(-) 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']