77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
import pytest
|
|
from core.builtin_concepts import ReturnValueConcept, BuiltinConcepts, ParserResultConcept
|
|
from core.concept import Concept
|
|
from core.rule import Rule
|
|
from evaluators.OneSuccessEvaluator import OneSuccessEvaluator
|
|
|
|
from tests.TestUsingMemoryBasedSheerka import TestUsingMemoryBasedSheerka
|
|
|
|
|
|
def r(name, status=True, value=None):
|
|
if isinstance(value, Rule):
|
|
value = ParserResultConcept(name, value=value, try_parsed=value)
|
|
return ReturnValueConcept(name, status, value or name)
|
|
|
|
|
|
reduce_requested = ReturnValueConcept("some_name", True, Concept(key=BuiltinConcepts.REDUCE_REQUESTED))
|
|
|
|
|
|
class TestOneSuccessEvaluator(TestUsingMemoryBasedSheerka):
|
|
|
|
@pytest.mark.parametrize("return_values, expected", [
|
|
([r("evaluators.one success"), reduce_requested], True),
|
|
([r("evaluators.one success"), r("failed", False), r("failed", False), reduce_requested], True),
|
|
([r("evaluators.no reduce required"), r("failed", False), r("failed", False)], False),
|
|
([r("evaluators.no reduce required")], False),
|
|
([r("evaluators.failed", False), r("failed", False), r("failed", False), reduce_requested], False),
|
|
([r("evaluators.failed", False), r("not evaluator success"), reduce_requested], False),
|
|
([r("evaluators.success"), r("evaluators.another success"), reduce_requested], False),
|
|
([r("evaluators.one success"), r("other success"), r("failed", False), reduce_requested], True),
|
|
([r("evaluators.one success"), r("parsers.success"), reduce_requested], False),
|
|
([r("evaluators.one success"), r("parsers.success"), reduce_requested], False),
|
|
])
|
|
def test_i_can_match(self, return_values, expected):
|
|
context = self.get_context()
|
|
assert OneSuccessEvaluator().matches(context, return_values) == expected
|
|
|
|
def test_i_can_eval(self):
|
|
context = self.get_context()
|
|
|
|
return_values = [
|
|
r("evaluators.one success"),
|
|
r("failed", False),
|
|
r("failed", False),
|
|
reduce_requested
|
|
]
|
|
|
|
evaluator = OneSuccessEvaluator()
|
|
matches = evaluator.matches(context, return_values)
|
|
res = evaluator.eval(context, return_values)
|
|
|
|
assert matches
|
|
assert res.status
|
|
assert res.body == "evaluators.one success"
|
|
assert len(res.parents) == 4
|
|
|
|
def test_i_do_not_eat_the_other_success(self):
|
|
context = self.get_context()
|
|
|
|
not_a_parser_success = r("other success")
|
|
return_values = [
|
|
r("evaluators.one success"),
|
|
not_a_parser_success,
|
|
r("failed", False),
|
|
reduce_requested
|
|
]
|
|
|
|
evaluator = OneSuccessEvaluator()
|
|
matches = evaluator.matches(context, return_values)
|
|
res = evaluator.eval(context, return_values)
|
|
|
|
assert matches
|
|
assert res.status
|
|
assert res.body == "evaluators.one success"
|
|
assert len(res.parents) == 3
|
|
|
|
assert not_a_parser_success not in res.parents
|