Use after_execution mechanism for test asserts.

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',
      {