def test_multi_put_item(ignore_comments):
'''Check that multiple lines can be pushed back and will be returned
correctly in the specified order (actually the reverse of the
original). Test with and without ignoring comments.
'''
reader = FortranStringReader(FORTRAN_CODE, ignore_comments=ignore_comments)
orig_lines = []
while True:
orig_line = reader.get_item()
if not orig_line:
break
# Make sure our original lines are kept in reverse order.
orig_lines.insert(0, orig_line)
# Put back original lines in reverse order as that is what we
# would expect when processing and rolling back.
for line in orig_lines:
reader.put_item(line)
# Lines should now be returned in the correct order (so compare in
# reverse order with the original line list)
while True:
filo_line = reader.get_item()
if not filo_line:
break
assert filo_line == orig_lines.pop(-1)
assert not orig_lines
def test_put_item_include(ignore_comments):
'''Check that when a line that has been included via an include
statement is consumed it can be pushed back so it can be consumed
again. Test with and without ignoring comments.
'''
reader = FortranStringReader(FORTRAN_CODE, ignore_comments=ignore_comments)
while True:
orig_line = reader.get_item()
if not orig_line:
break
reader.put_item(orig_line)
fifo_line = reader.get_item()
assert fifo_line == orig_line
def test_get_item(ignore_comments):
'''Check the get_item() function works as expected. Test with and
without comments being ignored.
'''
if ignore_comments:
expected = EXPECTED_CODE
else:
expected = ("program test\n"
"! prog comment 1\n"
"print *, 'Hello'\n"
"! prog comment 2\n"
"end program")
reader = FortranStringReader(FORTRAN_CODE, ignore_comments=ignore_comments)
for expected_line in expected.split("\n"):
output_line = reader.get_item()
assert expected_line in output_line.line
assert not reader.get_item()
