def test_subroutines1():
source = """\
subroutine sub1(a, b)
integer, intent(in) :: a
integer, intent(out) :: b
b = a + 1
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
def test_unknown_type2b():
source = """\
subroutine sub1(a)
integer, intent(in) :: a
b = 1
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
with pytest.raises(UndeclaredVariableError):
annotate_tree(tree, symbol_table)
def test_logical3():
source = """\
subroutine sub1(a, b)
integer, intent(in) :: a, b
integer :: c
if (a == b) c = 1
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
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_logical4():
source = """\
subroutine sub1(a, b)
integer, intent(in) :: a, b
integer :: c
if (a) c = 1
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
with pytest.raises(TypeMismatch):
annotate_tree(tree, symbol_table)
def test_type7():
source = """\
subroutine sub1(a, b)
real, intent(in) :: a
real, intent(in) :: b
a = (b + a)*a
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
assert tree.body[0]._type == Real()
def test_arrays1():
source = """\
subroutine sub1()
integer :: a(3), i
a(1) = 1
i = 2
a(2) = i
a(i) = i+1
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
def test_logical1():
source = """\
subroutine sub1(a, b)
integer, intent(in) :: a
integer, intent(in) :: b
logical :: r
r = (a == b)
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
assert tree.body[0]._type == Logical()
def test_arrays2():
source = """\
subroutine sub1()
integer :: a(3)
real :: r
a(1) = 1
r = 2
a(r) = 2
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
with pytest.raises(TypeMismatch):
annotate_tree(tree, symbol_table)
def test_logical5():
source = """\
subroutine sub1(a, b)
integer, intent(in) :: a, b
integer :: c
if (a == b) then
d = 1
end if
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
with pytest.raises(UndeclaredVariableError):
annotate_tree(tree, symbol_table)
def test_arrays4():
source = """\
subroutine sub1()
integer :: a(3), b(4), i
a(1) = b(2)
a(i) = b(2)
a(1) = b(2) + 1
a(1) = 1 + b(2)
a(1) = b(i) + 1
a(1) = b(i) + b(1)
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
def test_unknown_type3():
source = """\
module test
implicit none
contains
subroutine sub1(a)
integer, intent(in) :: a
a = (1+a)**2
end subroutine
end module
"""
tree = src_to_ast(source)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
def test_arrays3():
# Here we test assigning the wrong type to an array. Since only real and
# integer types are implemented for now, just test that real cannot be
# assinged to an integer. Later we can change this to, say, a character.
source = """\
subroutine sub1()
integer :: a(3), i
real :: r
i = 1
a(i) = r
end subroutine
"""
tree = src_to_ast(source, False)
symbol_table = create_symbol_table(tree)
with pytest.raises(TypeMismatch):
annotate_tree(tree, symbol_table)
def test_unknown_type2():
source = """\
module test
implicit none
contains
subroutine sub1(a)
integer, intent(in) :: a
a = (1+b)**2
end subroutine
end module
"""
tree = src_to_ast(source)
symbol_table = create_symbol_table(tree)
with pytest.raises(UndeclaredVariableError):
annotate_tree(tree, symbol_table)
def test_type1():
source = """\
module test
implicit none
contains
subroutine sub1(a, b)
integer, intent(in) :: a, b
a = b
end subroutine
end module
"""
tree = src_to_ast(source)
symbol_table = create_symbol_table(tree)
annotate_tree(tree, symbol_table)
assert tree.contains[0].body[0]._type == Integer()
def test_type6():
source = """\
module test
implicit none
contains
subroutine sub1(a, b)
integer, intent(in) :: a
real, intent(in) :: b
a = (b + 1)*a*5
end subroutine
end module
"""
tree = src_to_ast(source)
symbol_table = create_symbol_table(tree)
with pytest.raises(TypeMismatch):
annotate_tree(tree, symbol_table)
def get_num(src):
numexpr = src_to_ast(src, translation_unit=False)
symbol_table = create_symbol_table(numexpr)
annotate_tree(numexpr, symbol_table)
return numexpr.o