def test_indent():
codelines = ('subroutine test(a)\n'
'integer :: a, i, j\n'
'\n'
'do\n'
'do \n'
'do j = 1, 5\n'
'if (a>b) then\n'
'if(b>0) then\n'
'a = 3\n'
'donot_indent_me = 2\n'
'do_not_indent_me_either = 2\n'
'ifIam_indented_something_went_wrong = 2\n'
'if_I_am_indented_something_went_wrong = 2\n'
'end should not be unindented here\n'
'end if\n'
'endif\n'
'end do\n'
'end do\n'
'enddo\n'
'end subroutine\n'
'\n'
'subroutine test2(a)\n'
'integer :: a\n'
'do\n'
'a = a + 1\n'
'end do \n'
'end subroutine\n')
expected = ('subroutine test(a)\n'
'integer :: a, i, j\n'
'\n'
'do\n'
' do \n'
' do j = 1, 5\n'
' if (a>b) then\n'
' if(b>0) then\n'
' a = 3\n'
' donot_indent_me = 2\n'
' do_not_indent_me_either = 2\n'
' ifIam_indented_something_went_wrong = 2\n'
' if_I_am_indented_something_went_wrong = 2\n'
' end should not be unindented here\n'
' end if\n'
' endif\n'
' end do\n'
' end do\n'
'enddo\n'
'end subroutine\n'
'\n'
'subroutine test2(a)\n'
'integer :: a\n'
'do\n'
' a = a + 1\n'
'end do \n'
'end subroutine\n')
p = FCodePrinter({'source_format': 'free'})
result = p.indent_code(codelines)
assert result == expected
def test_indent():
codelines = ("subroutine test(a)\n"
"integer :: a, i, j\n"
"\n"
"do\n"
"do \n"
"do j = 1, 5\n"
"if (a>b) then\n"
"if(b>0) then\n"
"a = 3\n"
"donot_indent_me = 2\n"
"do_not_indent_me_either = 2\n"
"ifIam_indented_something_went_wrong = 2\n"
"if_I_am_indented_something_went_wrong = 2\n"
"end should not be unindented here\n"
"end if\n"
"endif\n"
"end do\n"
"end do\n"
"enddo\n"
"end subroutine\n"
"\n"
"subroutine test2(a)\n"
"integer :: a\n"
"do\n"
"a = a + 1\n"
"end do \n"
"end subroutine\n")
expected = ("subroutine test(a)\n"
"integer :: a, i, j\n"
"\n"
"do\n"
" do \n"
" do j = 1, 5\n"
" if (a>b) then\n"
" if(b>0) then\n"
" a = 3\n"
" donot_indent_me = 2\n"
" do_not_indent_me_either = 2\n"
" ifIam_indented_something_went_wrong = 2\n"
" if_I_am_indented_something_went_wrong = 2\n"
" end should not be unindented here\n"
" end if\n"
" endif\n"
" end do\n"
" end do\n"
"enddo\n"
"end subroutine\n"
"\n"
"subroutine test2(a)\n"
"integer :: a\n"
"do\n"
" a = a + 1\n"
"end do \n"
"end subroutine\n")
p = FCodePrinter({"source_format": "free"})
result = p.indent_code(codelines)
assert result == expected
def _indent_code(self, codelines):
p = FCodePrinter({'source_format': 'free', 'human': False})
return p.indent_code(codelines)
def test_indent():
codelines = (
'subroutine test(a)\n'
'integer :: a, i, j\n'
'\n'
'do\n'
'do \n'
'do j = 1, 5\n'
'if (a>b) then\n'
'if(b>0) then\n'
'a = 3\n'
'donot_indent_me = 2\n'
'do_not_indent_me_either = 2\n'
'ifIam_indented_something_went_wrong = 2\n'
'if_I_am_indented_something_went_wrong = 2\n'
'end should not be unindented here\n'
'end if\n'
'endif\n'
'end do\n'
'end do\n'
'enddo\n'
'end subroutine\n'
'\n'
'subroutine test2(a)\n'
'integer :: a\n'
'do\n'
'a = a + 1\n'
'end do \n'
'end subroutine\n'
)
expected = (
'subroutine test(a)\n'
'integer :: a, i, j\n'
'\n'
'do\n'
' do \n'
' do j = 1, 5\n'
' if (a>b) then\n'
' if(b>0) then\n'
' a = 3\n'
' donot_indent_me = 2\n'
' do_not_indent_me_either = 2\n'
' ifIam_indented_something_went_wrong = 2\n'
' if_I_am_indented_something_went_wrong = 2\n'
' end should not be unindented here\n'
' end if\n'
' endif\n'
' end do\n'
' end do\n'
'enddo\n'
'end subroutine\n'
'\n'
'subroutine test2(a)\n'
'integer :: a\n'
'do\n'
' a = a + 1\n'
'end do \n'
'end subroutine\n'
)
p = FCodePrinter({'source_format': 'free'})
result = p.indent_code(codelines)
assert result == expected
def _indent_code(self, codelines):
p = FCodePrinter({'source_format': 'free', 'human': False})
return p.indent_code(codelines)
