From 02f909269a380697fe39e5d82a6021d441ca3cbc Mon Sep 17 00:00:00 2001
From: kenkeiras <kenkeiras@codigoparallevar.com>
Date: Wed, 24 May 2017 21:09:23 +0200
Subject: [PATCH] Use after_execution mechanism for test asserts.

---
 naive-nlu/tree_nlu/tests/gac_100.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/naive-nlu/tree_nlu/tests/gac_100.py b/naive-nlu/tree_nlu/tests/gac_100.py
index 29d5ead..b5021ec 100644
--- a/naive-nlu/tree_nlu/tests/gac_100.py
+++ b/naive-nlu/tree_nlu/tests/gac_100.py
@@ -1,5 +1,11 @@
 from ..knowledge_base import KnowledgeBase
 
+def _assert(args):
+    assert(args)
+
+def _assert_msg(args, msg):
+    assert(args, msg)
+
 examples = [
     ('full_example',
      {
@@ -8,6 +14,9 @@ examples = [
          "parsed": ("question",
                     ("exists-property-with-value", 'icecream', 'cold')),
          "answer": True,
+         "after_execution": [(
+             lambda knowledge: _assert('cold' in knowledge.knowledge['icecream']['property'])
+         ),],
      }),
     ('full_example',
      {
@@ -16,6 +25,9 @@ examples = [
          "parsed": ("question",
                     ("pertenence-to-group", 'earth', 'planet')),
          "answer": True,
+         "after_execution": [(
+             lambda knowledge: _assert('planet' in knowledge.knowledge['earth']['groups'])
+         ),],
      }),
     ('full_example',
      {
@@ -24,6 +36,9 @@ examples = [
          "parsed": ("question",
                     ("pertenence-to-group", 'green', 'color')),
          "answer": True,
+         "after_execution": [(
+             lambda knowledge: _assert('color' in knowledge.knowledge['green']['groups'])
+         ),],
      }),
     ('full_example',
      {
@@ -32,6 +47,9 @@ examples = [
          "parsed": ("question",
                     ("has-capacity", 'plane', 'fly')),
          "answer": True,
+         "after_execution": [(
+             lambda knowledge: _assert('fly' in knowledge.knowledge['plane']['capacities'])
+         ),],
      }),
     ('full_example',
      {
@@ -40,6 +58,9 @@ examples = [
          "parsed": ("question",
                     ("implies", 'summer', 'hot')),
          "answer": True,
+         "after_execution": [(
+             lambda knowledge: _assert('hot' in knowledge.knowledge['summer']['implications'])
+         ),],
      }),
     ('full_example',
      {