def test_parse_func_app_no_args():
sc = Scanner('#<func_app_no_args>', "foo()", True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, AppExpression))
assert (isinstance(e.operator, VarRefExpression))
assert (e.operator.name == 'foo')
assert (len(e.operands) == 0)
def test_parse_simple_index():
sc = Scanner("#<simple_index>", 'foo[2]', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, IndexExpression))
assert (isinstance(e.expression, VarRefExpression))
assert (e.expression.name == 'foo')
assert (isinstance(e.index, ConstExpression))
assert (e.index.value == 2)
def test_parse_simple_member_access():
sc = Scanner('#<member_access>', 'foo.bar', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, MemberExpression))
assert (isinstance(e.expression, VarRefExpression))
assert (e.expression.name == 'foo')
assert (len(e.members) == 1)
assert (e.members[0] == 'bar')
def test_parse_func_app_one_arg():
sc = Scanner('#<func_app_one_arg>', "foo(bar)", True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, AppExpression))
assert (isinstance(e.operator, VarRefExpression))
assert (e.operator.name == 'foo')
assert (len(e.operands) == 1)
assert (isinstance(e.operands[0], VarRefExpression))
assert (e.operands[0].name == 'bar')
def test_parse_simple_add():
sc = Scanner('#<simple_add>', 'a + b', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, AppExpression))
assert (len(e.operands) == 2)
assert (isinstance(e.operands[0], VarRefExpression))
assert (e.operands[0].name == 'a')
assert (isinstance(e.operands[1], VarRefExpression))
assert (e.operands[1].name == 'b')
def test_parse_simple_slice():
sc = Scanner("#<simple_slice>", 'foo[1:2]', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, SliceExpression))
assert (isinstance(e.expression, VarRefExpression))
assert (e.expression.name == 'foo')
assert (isinstance(e.min, ConstExpression))
assert (e.min.value == 1)
assert (isinstance(e.max, ConstExpression))
assert (e.max.value == 2)
def test_parse_binary_operator_precedence4():
sc = Scanner('#<binary_operator_precedence4>', 'a ** b ** c', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, AppExpression))
assert (len(e.operands) == 2)
arg1 = e.operands[0]
assert (isinstance(arg1, VarRefExpression))
assert (arg1.name == 'a')
arg2 = e.operands[1]
assert (isinstance(arg2, AppExpression))
assert (arg2.operands[0].name == 'b')
assert (arg2.operands[1].name == 'c')
def test_parse_nested_add():
sc = Scanner('#<nested_add>', '(a + b) + c', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, AppExpression))
assert (len(e.operands) == 2)
arg1 = e.operands[0]
assert (isinstance(arg1, AppExpression))
assert (arg1.operands[0].name == 'a')
assert (arg1.operands[1].name == 'b')
arg2 = e.operands[1]
assert (isinstance(arg2, VarRefExpression))
assert (arg2.name == 'c')
def test_parse_func_app_many_args():
sc = Scanner('#<func_app_many_args>', "foo(bar, 1, 2, 3)", True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, AppExpression))
assert (isinstance(e.operator, VarRefExpression))
assert (e.operator.name == 'foo')
assert (len(e.operands) == 4)
assert (isinstance(e.operands[0], VarRefExpression))
assert (e.operands[0].name == 'bar')
assert (isinstance(e.operands[1], ConstExpression))
assert (e.operands[1].value == 1)
assert (isinstance(e.operands[2], ConstExpression))
assert (e.operands[2].value == 2)
assert (isinstance(e.operands[3], ConstExpression))
assert (e.operands[3].value == 3)
def test_parse_complex_expression():
sc = Scanner('#<member_access>', '(1 + 2).bar.baz.bax', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, MemberExpression))
assert (isinstance(e.expression, AppExpression))
assert (isinstance(e.expression.operator, VarRefExpression))
assert (e.expression.operator.name == '+')
assert (len(e.expression.operands) == 2)
assert (isinstance(e.expression.operands[0], ConstExpression))
assert (e.expression.operands[0].value == 1)
assert (isinstance(e.expression.operands[1], ConstExpression))
assert (e.expression.operands[1].value == 2)
assert (len(e.members) == 3)
assert (e.members[0] == 'bar')
assert (e.members[1] == 'baz')
assert (e.members[2] == 'bax')
def test_parse_complex_access():
sc = Scanner("#<complex_access>", 'foo[1:2][3].bar[4]', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, IndexExpression))
assert (isinstance(e.index, ConstExpression))
assert (e.index.value == 4)
e1 = e.expression
assert (isinstance(e1, MemberExpression))
assert (e1.members[0] == 'bar')
e2 = e1.expression
assert (isinstance(e2, IndexExpression))
assert (isinstance(e2.index, ConstExpression))
assert (e2.index.value == 3)
e3 = e2.expression
assert (isinstance(e3, SliceExpression))
assert (isinstance(e3.min, ConstExpression))
assert (e3.min.value == 1)
assert (isinstance(e3.max, ConstExpression))
assert (e3.max.value == 2)
e4 = e3.expression
assert (isinstance(e4, VarRefExpression))
assert (e4.name == 'foo')
def test_parse_not_operator():
sc = Scanner('#<not_operator>', 'not a', True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, AppExpression))
def test_parse_string_lit():
sc = Scanner('#<string_lit>', "'Hello!'", True)
p = Parser(sc)
e = p.parse_expression()
assert (isinstance(e, ConstExpression))
assert (e.value == 'Hello!')