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