From 6fb1e1e6495871d36b325de036856ddac9f2e4ca Mon Sep 17 00:00:00 2001
From: kenkeiras <kenkeiras@codigoparallevar.com>
Date: Sun, 15 Apr 2018 20:13:45 +0200
Subject: [PATCH] Replace debugging prints by session logs.

---
 naive-nlu/tree_nlu/knowledge_base.py     | 1 -
 naive-nlu/tree_nlu/tests/tokenization.py | 4 ++--
 naive-nlu/tree_nlu/utils/tokenization.py | 9 +++++++--
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/naive-nlu/tree_nlu/knowledge_base.py b/naive-nlu/tree_nlu/knowledge_base.py
index f8cfa99..218b09a 100644
--- a/naive-nlu/tree_nlu/knowledge_base.py
+++ b/naive-nlu/tree_nlu/knowledge_base.py
@@ -128,7 +128,6 @@ class KnowledgeBase(object):
         knowledge_before = copy.deepcopy(self.knowledge)
         with session().log("Process: {}".format(row)):
             tokens = self.tokenize(row)
-            print(tokens)
 
             fit = parsing.get_fit(self, tokens)
             if fit is None:
diff --git a/naive-nlu/tree_nlu/tests/tokenization.py b/naive-nlu/tree_nlu/tests/tokenization.py
index 4b91dae..7e93d59 100644
--- a/naive-nlu/tree_nlu/tests/tokenization.py
+++ b/naive-nlu/tree_nlu/tests/tokenization.py
@@ -65,8 +65,8 @@ def main():
             with session().log(example['text']):
                 tokens = list(knowledge.tokenize(example['text']))
 
-                print(tokens)
-                print(example['tokens'])
+                session().log('Expected “{}”, found “{}”'
+                            .format(tokens, example['tokens']))
                 assert example['tokens'] == tokens
 
         else:
diff --git a/naive-nlu/tree_nlu/utils/tokenization.py b/naive-nlu/tree_nlu/utils/tokenization.py
index 9b9ee11..b763584 100644
--- a/naive-nlu/tree_nlu/utils/tokenization.py
+++ b/naive-nlu/tree_nlu/utils/tokenization.py
@@ -1,3 +1,7 @@
+from ..session.org_mode import (
+    global_session as session,
+)
+
 BASIC_TOKENIZATION_EXAMPLES = (
     ({
         "text": 'cat',
@@ -15,5 +19,6 @@ BASIC_TOKENIZATION_EXAMPLES = (
 
 
 def train_basic_tokenization(knowledge_base):
-    for example in BASIC_TOKENIZATION_EXAMPLES:
-        knowledge_base.train_tokenizer(example)
+    with session().log('Training basic tokenization'):
+        for example in BASIC_TOKENIZATION_EXAMPLES:
+            knowledge_base.train_tokenizer(example)