60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
import pytest
|
|
|
|
from core.builtin_concepts import ReturnValueConcept, BuiltinConcepts
|
|
from core.concept import Concept
|
|
from core.sheerka import Sheerka, ExecutionContext
|
|
from evaluators.EvalEvaluator import EvalEvaluator
|
|
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(value, status=True):
|
|
return ReturnValueConcept("some_name", status, value)
|
|
|
|
|
|
eval_requested = ReturnValueConcept("some_name", True, Concept(key=BuiltinConcepts.CONCEPT_EVAL_REQUESTED))
|
|
|
|
|
|
def test_i_can_match_and_eval():
|
|
context = get_context()
|
|
|
|
to_eval1 = ReturnValueConcept("some_name", True, Concept(name="2", body="to eval"))
|
|
to_eval2 = ReturnValueConcept("some_name", True, Concept(name="3", body="also to eval"))
|
|
|
|
return_values = [
|
|
ReturnValueConcept("some_name", True, "not to eval"),
|
|
ReturnValueConcept("some_name", True, Concept(name="not to eval")),
|
|
ReturnValueConcept("some_name", False, Concept(name="1", body="not to eval")),
|
|
to_eval1,
|
|
to_eval2,
|
|
eval_requested
|
|
]
|
|
|
|
evaluator = EvalEvaluator()
|
|
assert evaluator.matches(context, return_values)
|
|
|
|
evaluated = evaluator.eval(context, return_values)
|
|
assert len(evaluated) == 2
|
|
assert evaluated[0].value == to_eval1.body.body
|
|
assert evaluated[0].parents == [to_eval1, eval_requested]
|
|
|
|
assert evaluated[1].value == to_eval2.body.body
|
|
assert evaluated[1].parents == [to_eval2, eval_requested]
|
|
|
|
|
|
@pytest.mark.parametrize("return_values, expected", [
|
|
([r(Concept("foo", body="bar")), eval_requested], True),
|
|
([r(Concept("status is false", body="bar"), False), eval_requested], False),
|
|
([r("string_value"), eval_requested], False),
|
|
([r(Concept("no body")), eval_requested], False),
|
|
([r(Concept("eval requested missing", body="bar"))], False),
|
|
])
|
|
def test_i_cannot_match_if_eval_request_is_not_present(return_values, expected):
|
|
context = get_context()
|
|
assert EvalEvaluator().matches(context, return_values) == expected
|