def test_dump_expr():
assert dump(src_to_ast("1+1", False)) == \
"BinOp(left=Num(n='1'), op=Add(), right=Num(n='1'))"
assert dump(src_to_ast("1+x", False)) == \
"BinOp(left=Num(n='1'), op=Add(), right=Name(id='x'))"
assert dump(src_to_ast("(x+y)**2", False)) == \
"BinOp(left=BinOp(left=Name(id='x'), op=Add(), " \
"right=Name(id='y')), op=Pow(), right=Num(n='2'))"
def test_dump_programs():
assert dump(src_to_ast("""\
program a
integer :: b
b = 1
end program
""")) == "Program(name='a', use=[], decl=[Declaration(vars=[decl(sym='b', sym_type='integer', dims=[], attrs=[], initializer=None)])], body=[Assignment(target=Name(id='b'), value=Num(n='1'))], contains=[])"
def test_dump_subroutines():
assert dump(
src_to_ast(
"""\
subroutine f
integer :: a, b
b = (1+2+a)*3
end subroutine
""", False)
) == "Subroutine(name='f', args=[], use=[], decl=[Declaration(vars=[decl(sym='a', sym_type='integer', dims=[], attrs=[], initializer=None), decl(sym='b', sym_type='integer', dims=[], attrs=[], initializer=None)])], body=[Assignment(target=Name(id='b'), value=BinOp(left=BinOp(left=BinOp(left=Num(n='1'), op=Add(), right=Num(n='2')), op=Add(), right=Name(id='a')), op=Mul(), right=Num(n='3')))], contains=[])"
assert dump(
src_to_ast(
"""\
subroutine f(x, y)
integer, intent(out) :: x, y
x = 1
end subroutine
""", False)
) == "Subroutine(name='f', args=[arg(arg='x'), arg(arg='y')], use=[], decl=[Declaration(vars=[decl(sym='x', sym_type='integer', dims=[], attrs=[Attribute(name='intent', args=[attribute_arg(arg='out')])], initializer=None), decl(sym='y', sym_type='integer', dims=[], attrs=[Attribute(name='intent', args=[attribute_arg(arg='out')])], initializer=None)])], body=[Assignment(target=Name(id='x'), value=Num(n='1'))], contains=[])"
def test_dump():
source = """\
subroutine sub1(a, b)
integer, intent(in) :: a
integer, intent(in) :: b
a = (b + 1)*a*5
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
assert dump(tree, include_type=True) == "Subroutine(name='sub1', args=[arg(arg='a'), arg(arg='b')], use=[], decl=[Declaration(vars=[decl(sym='a', sym_type='integer', dims=[], attrs=[Attribute(name='intent', args=[attribute_arg(arg='in')])], initializer=None)]), Declaration(vars=[decl(sym='b', sym_type='integer', dims=[], attrs=[Attribute(name='intent', args=[attribute_arg(arg='in')])], initializer=None)])], body=[Assignment(target=Name(id='a'), value=BinOp(left=BinOp(left=BinOp(left=Name(id='b', type=Integer()), op=Add(), right=Num(n='1', type=Integer()), type=Integer()), op=Mul(), right=Name(id='a', type=Integer()), type=Integer()), op=Mul(), right=Num(n='5', type=Integer()), type=Integer()), type=Integer())], contains=[])"
def test_dump_statements():
assert dump(src_to_ast("if (x == 1) stop\n", False)) == \
"If(test=Compare(left=Name(id='x'), op=Eq(), right=Num(n='1')), " \
"body=[Stop(code=None)], orelse=[])"
assert dump(src_to_ast("x == 1\n", False)) == \
"Compare(left=Name(id='x'), op=Eq(), right=Num(n='1'))"
from lfortran.ast import src_to_ast, dump, print_tree
from lfortran.ast.tests.test_parser import to_tuple
source = open("examples/expr2.f90").read()
t = src_to_ast(source)
print(source)
print()
print("AST (tuple):")
print(to_tuple(t))
print()
print("AST (dump):")
print(dump(t))
print()
print("AST (dump_tree):")
print_tree(t)
