50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
import pytest
|
|
|
|
from core.builtin_concepts import BuiltinConcepts, ReturnValueConcept, UserInputConcept
|
|
from core.concept import Concept
|
|
from evaluators.PrepareEvalEvaluator import PrepareEvalEvaluator
|
|
|
|
from tests.TestUsingMemoryBasedSheerka import TestUsingMemoryBasedSheerka
|
|
|
|
r = ReturnValueConcept
|
|
|
|
|
|
class TestPrepareEvalEvaluator(TestUsingMemoryBasedSheerka):
|
|
|
|
@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(self, ret_val, expected):
|
|
context = self.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(self, ret_val, expected):
|
|
context = self.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)
|