70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
import ast
|
|
|
|
import pytest
|
|
|
|
from core.builtin_concepts import ParserResultConcept
|
|
from core.sheerka import Sheerka, ExecutionContext
|
|
from core.tokenizer import Tokenizer
|
|
from parsers.PythonParser import PythonNode, PythonParser, PythonErrorNode
|
|
from sdp.sheerkaDataProvider import Event
|
|
|
|
|
|
def get_context():
|
|
sheerka = Sheerka(skip_builtins_in_db=True)
|
|
sheerka.initialize("mem://")
|
|
return ExecutionContext("test", Event(), sheerka)
|
|
|
|
|
|
@pytest.mark.parametrize("text, expected", [
|
|
("1+1", PythonNode("1+1", ast.parse("1+1", mode="eval"))),
|
|
("a=10", PythonNode("a=10", ast.parse("a=10", mode="exec"))),
|
|
])
|
|
def test_i_can_parse_a_simple_expression(text, expected):
|
|
parser = PythonParser()
|
|
res = parser.parse(get_context(), text)
|
|
|
|
assert res.status
|
|
assert res.who == parser.name
|
|
assert isinstance(res.value, ParserResultConcept)
|
|
assert res.value.value == expected
|
|
|
|
|
|
@pytest.mark.parametrize("text, expected", [
|
|
("1+1", PythonNode("1+1", ast.parse("1+1", mode="eval"))),
|
|
("a=10", PythonNode("a=10", ast.parse("a=10", mode="exec"))),
|
|
])
|
|
def test_i_can_parse_from_tokens(text, expected):
|
|
parser = PythonParser()
|
|
tokens = list(Tokenizer(text))
|
|
res = parser.parse(get_context(), tokens)
|
|
|
|
assert res.status
|
|
assert res.who == parser.name
|
|
assert isinstance(res.value, ParserResultConcept)
|
|
assert res.value.value == expected
|
|
|
|
|
|
def test_i_can_detect_error():
|
|
text = "1+"
|
|
|
|
parser = PythonParser()
|
|
res = parser.parse(get_context(), text)
|
|
|
|
assert not res.status
|
|
assert res.who == parser.name
|
|
assert isinstance(res.value, ParserResultConcept)
|
|
assert isinstance(res.value.value[0], PythonErrorNode)
|
|
assert isinstance(res.value.value[0].exception, SyntaxError)
|
|
|
|
|
|
def test_i_can_parse_a_concept():
|
|
text = "c:concept_name: + 1"
|
|
|
|
parser = PythonParser()
|
|
res = parser.parse(get_context(), text)
|
|
|
|
assert res
|
|
assert res.value.value == PythonNode(
|
|
"c:concept_name: + 1",
|
|
ast.parse("__C__concept_name__C__+1", mode="eval"))
|