From 58fb9fb1883a3ecdff28b454c4393cd9a60e79c9 Mon Sep 17 00:00:00 2001
From: kenkeiras <kenkeiras@codigoparallevar.com>
Date: Sat, 30 Sep 2017 00:57:19 +0200
Subject: [PATCH] Add extended gac tests.

---
 naive-nlu/tree_nlu/test.py                |  2 ++
 naive-nlu/tree_nlu/tests/gac_extension.py | 21 +++++++++++++++++++++
 2 files changed, 23 insertions(+)
 create mode 100644 naive-nlu/tree_nlu/tests/gac_extension.py

diff --git a/naive-nlu/tree_nlu/test.py b/naive-nlu/tree_nlu/test.py
index 49ab053..3a67370 100644
--- a/naive-nlu/tree_nlu/test.py
+++ b/naive-nlu/tree_nlu/test.py
@@ -2,12 +2,14 @@ import traceback
 import logging
 from .tests import basic
 from .tests import gac_100
+from .tests import gac_extension
 
 logging.getLogger().setLevel(logging.ERROR)
 
 tests = (
     ("basic", basic),
     ("gac 100", gac_100),
+    ("gac+", gac_extension),
 )
 
 def main():
diff --git a/naive-nlu/tree_nlu/tests/gac_extension.py b/naive-nlu/tree_nlu/tests/gac_extension.py
new file mode 100644
index 0000000..8a9e8c0
--- /dev/null
+++ b/naive-nlu/tree_nlu/tests/gac_extension.py
@@ -0,0 +1,21 @@
+from ..knowledge_base import KnowledgeBase
+
+from . import gac_100
+
+
+def ask_then_learn_test(knowledge: KnowledgeBase):
+    ret, _, _ = knowledge.process("is icecream blue?")
+    assert(ret is False)
+
+    ret, _, _ = knowledge.process("icecream is blue")
+
+    ret, _, _ = knowledge.process("is icecream blue?")
+    assert(ret is True)
+
+    return knowledge
+
+
+def main():
+    knowledge = gac_100.main()
+
+    knowledge = ask_then_learn_test(knowledge)