Replace debugging prints by session logs.

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:

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:

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)