56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
import pytest
|
|
|
|
from core.builtin_concepts import BuiltinConcepts, ReturnValueConcept, UserInputConcept
|
|
from core.concept import Concept
|
|
from core.sheerka import Sheerka, ExecutionContext
|
|
from evaluators.PrepareEvalEvaluator import PrepareEvalEvaluator
|
|
from sdp.sheerkaDataProvider import Event
|
|
|
|
|
|
def get_context():
|
|
sheerka = Sheerka(skip_builtins_in_db=True)
|
|
sheerka.initialize("mem://")
|
|
return ExecutionContext("test", Event(), sheerka)
|
|
|
|
|
|
r = ReturnValueConcept
|
|
|
|
|
|
@pytest.mark.parametrize("ret_val, expected", [
|
|
(r("name", True, UserInputConcept("eval 1 + 1")), True),
|
|
(r("name", True, UserInputConcept(" eval 1 + 1")), True),
|
|
(r("name", True, UserInputConcept("eval")), False),
|
|
(r("name", True, UserInputConcept("1+1")), False),
|
|
(r("name", True, UserInputConcept("")), False),
|
|
(r("name", True, UserInputConcept("eval ")), False),
|
|
(r("name", True, UserInputConcept([])), False),
|
|
(r("name", True, Concept("foo")), False),
|
|
(r("name", True, "not a concept"), False),
|
|
(r("name", False, UserInputConcept("eval 1 + 1")), False),
|
|
(r("name", False, UserInputConcept(" eval 1 + 1")), False),
|
|
])
|
|
def test_i_can_match(ret_val, expected):
|
|
context = get_context()
|
|
assert PrepareEvalEvaluator().matches(context, ret_val) == expected
|
|
|
|
|
|
@pytest.mark.parametrize("ret_val, expected", [
|
|
(r("name", True, UserInputConcept("eval 1 + 1")), "1 + 1"),
|
|
(r("name", True, UserInputConcept(" eval 1 + 1")), "1 + 1"),
|
|
])
|
|
def test_i_can_eval(ret_val, expected):
|
|
context = get_context()
|
|
sheerka = context.sheerka
|
|
|
|
prepare_evaluator = PrepareEvalEvaluator()
|
|
prepare_evaluator.matches(context, ret_val)
|
|
res = prepare_evaluator.eval(context, ret_val)
|
|
|
|
assert len(res) == 2
|
|
assert res[0].status
|
|
assert sheerka.isinstance(res[0].body, BuiltinConcepts.USER_INPUT)
|
|
assert res[0].body.body == expected
|
|
|
|
assert res[1].status
|
|
assert sheerka.isinstance(res[1].body, BuiltinConcepts.CONCEPT_EVAL_REQUESTED)
|