You must now use 'eval' to get the body of a concept

2019-12-24 16:58:09 +01:00
parent 5c90b07e1a
commit 44e4b75cf8
37 changed files with 1003 additions and 383 deletions
@@ -0,0 +1,112 @@
+import pytest
+
+from core.builtin_concepts import ReturnValueConcept, BuiltinConcepts
+from core.concept import Concept
+from core.sheerka import Sheerka, ExecutionContext
+from evaluators.TooManySuccessEvaluator import TooManySuccessEvaluator
+from sdp.sheerkaDataProvider import Event
+
+
+def get_context():
+    sheerka = Sheerka(skip_builtins_in_db=True)
+    sheerka.initialize("mem://")
+    return ExecutionContext("test", Event(), sheerka)
+
+
+def r(name, status=True, value=None):
+    value = value or name
+    return ReturnValueConcept(name, status, value)
+
+
+reduce_requested = ReturnValueConcept("some_name", True, Concept(key=BuiltinConcepts.REDUCE_REQUESTED))
+
+
+@pytest.mark.parametrize("return_values, expected", [
+    ([r("evaluators.success1"), r("evaluators.success2"), reduce_requested], True),
+    ([r("evaluators.success1"), r("evaluators.success2"), r("other"), reduce_requested], True),
+    ([r("evaluators.success1"), r("evaluators.success2"), r("other", False), reduce_requested], True),
+    ([r("evaluators.a", value=Concept("c1", body="1")),
+      r("evaluators.a", value=Concept("c2", body="1")),
+      r("other"),
+      reduce_requested], True),
+    ([r("evaluators.a", value=Concept("c1", body="1")),
+      r("evaluators.a", value=Concept("c2", body="1")),
+      r("parsers.other", False),
+      reduce_requested], True),
+    ([r("evaluators.a", value=Concept("c1", body="1")),
+      r("evaluators.a", value=Concept("c2", body="1")),
+      r("parsers.other"),
+      reduce_requested], False),
+    ([r("evaluators.success1"), reduce_requested], False),
+    ([reduce_requested], False),
+    ([r("evaluators.success1"), r("evaluators.success2")], False),
+])
+def test_i_can_match(return_values, expected):
+    context = get_context()
+    assert TooManySuccessEvaluator().matches(context, return_values) == expected
+
+
+def test_i_can_eval():
+    context = get_context()
+
+    value1 = r("evaluators.a", value=Concept("c1", body="1"))
+    value2 = r("evaluators.a", value=Concept("c2", body="2"))
+    return_values = [
+        value1,
+        value2,
+        r("other", False),
+        reduce_requested
+    ]
+
+    evaluator = TooManySuccessEvaluator()
+    matches = evaluator.matches(context, return_values)
+    res = evaluator.eval(context, return_values)
+
+    assert matches
+    assert not res.status
+    assert context.sheerka.isinstance(res.body, BuiltinConcepts.TOO_MANY_SUCCESS)
+    assert res.body.body == [value1, value2]
+    assert len(res.parents) == 4
+
+
+def test_i_can_eval_when_same_success():
+    context = get_context()
+
+    return_values = [
+        r("evaluators.a", value=Concept("c1", body="1")),
+        r("evaluators.a", value=Concept("c2", body="1")),
+        r("other", False),
+        reduce_requested
+    ]
+
+    evaluator = TooManySuccessEvaluator()
+    matches = evaluator.matches(context, return_values)
+    res = evaluator.eval(context, return_values)
+
+    assert matches
+    assert res is None
+
+
+def test_other_success_are_not_reduced():
+    context = get_context()
+
+    value1 = r("evaluators.a", value=Concept("c1", body="1"))
+    value2 = r("evaluators.a", value=Concept("c2", body="2"))
+    other_success = r("other")
+    return_values = [
+        value1,
+        value2,
+        other_success,
+        reduce_requested
+    ]
+
+    evaluator = TooManySuccessEvaluator()
+    matches = evaluator.matches(context, return_values)
+    res = evaluator.eval(context, return_values)
+
+    assert matches
+    assert not res.status
+    assert context.sheerka.isinstance(res.body, BuiltinConcepts.TOO_MANY_SUCCESS)
+    assert res.body.body == [value1, value2]
+    assert len(res.parents) == 3
+    assert other_success not in res.parents